Update oqmd.py

* Add the allow_pickle option to np.load
Tony-Y · Dec 7, 2022 · 0c9c065 · 0c9c065
1 parent db3306a
commit 0c9c065
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/OQMD/oqmd.py b/OQMD/oqmd.py
@@ -55,9 +55,9 @@ def magnetization_per_atom(x):
 
 def load_materials(filepath):
     try:
-        data = np.load(filepath)['materials']
+        data = np.load(filepath, allow_pickle=True)['materials']
     except UnicodeError:
-        data = np.load(filepath, encoding='latin1')['materials']
+        data = np.load(filepath, allow_pickle=True, encoding='latin1')['materials']
     return data
 
 def get_prop(path):
@@ -85,7 +85,7 @@ def get_prop(path):
 graph_nodes = []
 graph_edges = []
 for path in graph_files:
-    data = np.load(path)
+    data = np.load(path, allow_pickle=True)
     graph_names += list(data['graph_names'])
     graph_nodes += list(data['graph_nodes'])
     graph_edges += list(data['graph_edges'])