v0.1.1

CompRhys · Aug 10, 2023 · ae0c2fd · ae0c2fd
1 parent e43685f
commit ae0c2fd
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 8 deletions.
diff --git a/examples/wrenformer/mat_bench/save_matbench_aflow_labels.py b/examples/wrenformer/mat_bench/save_matbench_aflow_labels.py
@@ -15,21 +15,21 @@
 """
 
 
-mbbm = MatbenchBenchmark()
+benchmark = MatbenchBenchmark()
 
-for idx, task in enumerate(mbbm.tasks, 1):
-    print(f"\n\n{idx}/{len(mbbm.tasks)}")
+for idx, task in enumerate(benchmark.tasks, 1):
+    print(f"\n\n{idx}/{len(benchmark.tasks)}")
     task.load()
     df: pd.DataFrame = task.df
 
     if "structure" in df:
-        df["composition"] = [x.formula for x in df.structure]
+        df["composition"] = [struct.formula for struct in df.structure]
         df["wyckoff"] = [
-            get_aflow_label_from_spglib(x)
-            for x in tqdm(df.structure, desc="Getting Aflow Wyckoff labels")
+            get_aflow_label_from_spglib(struct)
+            for struct in tqdm(df.structure, desc="Getting Aflow Wyckoff labels")
         ]
     elif "composition" in df:
-        df["composition"] = [x.formula for x in df.composition]
+        df["composition"] = [comp.formula for comp in df.composition]
     else:
         raise ValueError("No structure or composition column found")
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "aviary"
-version = "0.1.0"
+version = "0.1.1"
 description = "A collection of machine learning models for materials discovery"
 authors = [{ name = "Rhys Goodall", email = "[email protected]" }]
 readme = "README.md"