Update {Elastic,run,surf,Vasp,vasp}.py: To be compatible with Pymatgen

dpgen/auto_test/Elastic.py

@@ -221,7 +221,12 @@ def post_process(self, task_list):
                 kspacing = incar.get("KSPACING")
                 kgamma = incar.get("KGAMMA", False)
                 ret = vasp.make_kspacing_kpoints(poscar_start, kspacing, kgamma)
-                kp = Kpoints.from_string(ret)
+                pymgv = "old"
+                try:
+                    kp = Kpoints.from_string(ret)
+                except AttributeError:
+                    kp = Kpoints.from_str(ret)
+                    pymgv = "new"
                 if os.path.isfile("KPOINTS"):
                     os.remove("KPOINTS")
                 kp.write_file("KPOINTS")

dpgen/auto_test/VASP.py

@@ -194,7 +194,12 @@ def make_input_file(self, output_dir, task_type, task_param):
             os.symlink("../INCAR", "INCAR")
         os.chdir(cwd)
         ret = vasp.make_kspacing_kpoints(self.path_to_poscar, kspacing, kgamma)
-        kp = Kpoints.from_string(ret)
+        pymgv = "old"
+        try:
+            kp = Kpoints.from_string(ret)
+        except AttributeError:
+            kp = Kpoints.from_str(ret)
+            pymgv = "new"
         kp.write_file(os.path.join(output_dir, "KPOINTS"))
 
     def compute(self, output_dir):

dpgen/auto_test/lib/vasp.py

@@ -510,7 +510,12 @@ def make_vasp_kpoints_from_incar(work_dir, jdata):
     assert os.path.exists("INCAR")
     with open("INCAR") as fp:
         incar = fp.read()
-    standard_incar = incar_upper(Incar.from_string(incar))
+    pymgv = "old"
+    try:
+        standard_incar = incar_upper(Incar.from_string(incar))
+    except AttributeError:
+        standard_incar = incar_upper(Incar.from_str(incar))
+        pymgv = "new"
     if fp_aniso_kspacing is None:
         try:
             kspacing = standard_incar["KSPACING"]
@@ -533,6 +538,9 @@ def make_vasp_kpoints_from_incar(work_dir, jdata):
     assert os.path.exists("POSCAR")
     # make kpoints
     ret = make_kspacing_kpoints("POSCAR", kspacing, gamma)
-    kp = Kpoints.from_string(ret)
+    if pymgv == "new":
+        kp = Kpoints.from_str(ret)
+    elif pymgv == "old":
+        kp = Kpoints.from_string(ret)
     kp.write_file("KPOINTS")
     os.chdir(cwd)

dpgen/data/surf.py

@@ -410,7 +410,12 @@ def poscar_scale(poscar_in, poscar_out, scale):
     else:
         raise RuntimeError("Unknow poscar style at line 7: %s" % lines[7])
 
-    poscar = Poscar.from_string("".join(lines))
+    pymgv = "old"
+    try:
+        poscar = Poscar.from_string("".join(lines))
+    except AttributeError:
+        poscar = Poscar.from_str("".join(lines))
+        pymgv = "new"
     with open(poscar_out, "w") as fout:
         fout.write(poscar.get_string(direct=False))
 

dpgen/generator/run.py

@@ -2859,14 +2859,22 @@ def make_pwmat_input(jdata, filename):
 def make_vasp_incar_ele_temp(jdata, filename, ele_temp, nbands_esti=None):
     with open(filename) as fp:
         incar = fp.read()
-    incar = incar_upper(Incar.from_string(incar))
+    pymgv = "old"
+    try:
+        incar = incar_upper(Incar.from_string(incar))
+    except AttributeError:
+        incar = incar_upper(Incar.from_str(incar))
+        pymgv = "new"
     incar["ISMEAR"] = -1
     incar["SIGMA"] = ele_temp * pc.Boltzmann / pc.electron_volt
     incar.write_file("INCAR")
     if nbands_esti is not None:
         nbands = nbands_esti.predict(".")
         with open(filename) as fp:
-            incar = Incar.from_string(fp.read())
+            if pymgv == "new":
+                incar = Incar.from_str(fp.read())
+            elif pymgv == "old":
+                incar = Incar.from_string(fp.read())
         incar["NBANDS"] = nbands
         incar.write_file("INCAR")
 
@@ -2943,7 +2951,12 @@ def make_fp_vasp_kp(iter_index, jdata):
         assert os.path.exists("INCAR")
         with open("INCAR") as fp:
             incar = fp.read()
-        standard_incar = incar_upper(Incar.from_string(incar))
+        pymgv = "old"
+        try:
+            standard_incar = incar_upper(Incar.from_string(incar))
+        except AttributeError:
+            standard_incar = incar_upper(Incar.from_str(incar))
+            pymgv = "new"
         if fp_aniso_kspacing is None:
             try:
                 kspacing = standard_incar["KSPACING"]
@@ -2966,7 +2979,10 @@ def make_fp_vasp_kp(iter_index, jdata):
         assert os.path.exists("POSCAR")
         # make kpoints
         ret = make_kspacing_kpoints("POSCAR", kspacing, gamma)
-        kp = Kpoints.from_string(ret)
+        if pymgv == "new":
+            kp = Kpoints.from_str(ret)
+        elif pymgv == "old":
+            kp = Kpoints.from_string(ret)
         kp.write_file("KPOINTS")
         os.chdir(cwd)