v0.1.1

Author: janosh

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")
 

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 = "rhys.goodall@outlook.com" }]
 readme = "README.md"