ensure old bandstructure dicts produce numpy arrays and Spin objects

materialsproject · Dec 22, 2016 · b7533a5 · b7533a5
1 parent 6bb90b5
commit b7533a5
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/pymatgen/electronic_structure/bandstructure.py b/pymatgen/electronic_structure/bandstructure.py
@@ -863,10 +863,10 @@ def from_old_dict(cls, d):
                                                    orb])):
                                 ddddd.append(d['projections'][spin][i][j][
                                                 orb][l])
-                            dddd.append(ddddd)
-                        ddd.append(dddd)
-                    dd.append(ddd)
-                projections[spin] = dd
+                            dddd.append(np.array(ddddd))
+                        ddd.append(np.array(dddd))
+                    dd.append(np.array(ddd))
+                projections[Spin(int(spin))] = np.array(dd)
 
         return BandStructureSymmLine(
             d['kpoints'], {Spin(int(k)): d['bands'][k]