Fixed change of name for vasp input set generator.

cc4s · Sep 25, 2023 · bdb0a99 · bdb0a99
1 parent a576ea2
commit bdb0a99
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/pycc4s/workflows/sets/vasp.py b/src/pycc4s/workflows/sets/vasp.py
@@ -4,7 +4,7 @@
 """
 from dataclasses import dataclass
 
-from atomate2.vasp.sets.base import VaspInputSetGenerator
+from atomate2.vasp.sets.base import VaspInputGenerator
 from atomate2.vasp.sets.core import StaticSetGenerator
 from pymatgen.core.structure import Structure
 from pymatgen.io.vasp.outputs import Outcar, Vasprun
@@ -54,7 +54,7 @@ def get_incar_updates(
 
 
 @dataclass
-class NonSCFHFSetGenerator(VaspInputSetGenerator):
+class NonSCFHFSetGenerator(VaspInputGenerator):
     """Class to generate VASP non-self-consistent field Hartree-Fock input sets."""
 
     calc_type: str = "nonscf_hf"
@@ -110,7 +110,7 @@ def get_incar_updates(
 
 
 @dataclass
-class NonSCFMP2CBSSetGenerator(VaspInputSetGenerator):
+class NonSCFMP2CBSSetGenerator(VaspInputGenerator):
     """Class to generate VASP non-self-consistent field Hartree-Fock input sets."""
 
     calc_type: str = "nonscf_hf"
@@ -165,7 +165,7 @@ def get_incar_updates(
 
 
 @dataclass
-class NonSCFMP2NOsSetGenerator(VaspInputSetGenerator):
+class NonSCFMP2NOsSetGenerator(VaspInputGenerator):
     """Class to generate VASP non-self-consistent field Hartree-Fock input sets."""
 
     calc_type: str = "nonscf_hf"
@@ -220,7 +220,7 @@ def get_incar_updates(
 
 
 @dataclass
-class NonSCFHFNOsSetGenerator(VaspInputSetGenerator):
+class NonSCFHFNOsSetGenerator(VaspInputGenerator):
     """Class to generate VASP non-self-consistent field Hartree-Fock input sets."""
 
     calc_type: str = "nonscf_hf"
@@ -277,7 +277,7 @@ def get_incar_updates(
 
 
 @dataclass
-class VaspDumpCc4sFilesGenerator(VaspInputSetGenerator):
+class VaspDumpCc4sFilesGenerator(VaspInputGenerator):
     """Class to generate input set for dumping cc4s files."""
 
     calc_type: str = "vasp_cc4s_dump"