fix pymatgen update break

materialsvirtuallab · Oct 3, 2019 · b109355 · b109355
1 parent 3434a55
commit b109355
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/megnet/data/local_env.py b/megnet/data/local_env.py
@@ -36,11 +36,11 @@ def get_nn_info(self, structure, n):
 
         siw = []
         for nn in neighs_dists:
-            s, w = nn.site, nn.distance
-            siw.append({'site': s,
-                        'image': self._get_image(structure, s),
-                        'weight': w,
-                        'site_index': self._get_original_site(structure, s)})
+            neighbor, d, index, image  = nn[0], nn[1]
+            siw.append({'site': neighbor,
+                        'image': image 
+                        'weight': d,
+                        'site_index': index 
         return siw
 
     def get_all_nn_info(self, structure):