Merge pull request #153 from JiwooChloeLee/chloeLeeDOper

Chloe Lee Modify doper_example and doper.py files to a newer version
WMD-group · Aug 10, 2023 · c382095 · c382095
2 parents 18c0abc + aaabd06
commit c382095
Show file tree

Hide file tree

Showing 2 changed files with 135 additions and 147 deletions.
diff --git a/examples/Dopant_Prediction/doper_example.ipynb b/examples/Dopant_Prediction/doper_example.ipynb
diff --git a/smact/dopant_prediction/doper.py b/smact/dopant_prediction/doper.py
@@ -188,7 +188,7 @@ def plot_dopants(
             "p-type anion substitutions": self.p_type_an[:num_dopants],
         }
         for key, val in results.items():
-            dict_results = {utilities.parse_spec(x)[0]: y for x, y in val}
+            dict_results = {utilities.parse_spec(x)[0]: y for x, _, y in val}
             plotting.periodic_table_heatmap(
                 elemental_data=dict_results,
                 cmap="rainbow",