split pymatgen/io/vasp/sets.py

into sub-modules
materialsproject · Jun 5, 2024 · 1c51748 · 1c51748
1 parent 5c7889c
commit 1c51748
Show file tree

Hide file tree

Showing 7 changed files with 1,750 additions and 1,642 deletions.
diff --git a/pymatgen/io/vasp/sets/__init__.py b/pymatgen/io/vasp/sets/__init__.py
@@ -0,0 +1,40 @@
+from __future__ import annotations
+
+from pymatgen.io.vasp.sets.base import (
+    MODULE_DIR,
+    BadInputSetWarning,
+    DictSet,
+    UserPotcarFunctional,
+    VaspInputGenerator,
+    VaspInputSet,
+    _load_yaml_config,
+    batch_write_input,
+    get_structure_from_prev_run,
+    get_valid_magmom_struct,
+)
+from pymatgen.io.vasp.sets.lobster import LobsterSet
+from pymatgen.io.vasp.sets.matpes import MatPESStaticSet
+from pymatgen.io.vasp.sets.mit import MITMDSet, MITNEBSet, MITRelaxSet
+from pymatgen.io.vasp.sets.mp import (
+    MPAbsorptionSet,
+    MPHSEBSSet,
+    MPHSERelaxSet,
+    MPMDSet,
+    MPMetalRelaxSet,
+    MPNMRSet,
+    MPNonSCFSet,
+    MPRelaxSet,
+    MPScanRelaxSet,
+    MPScanStaticSet,
+    MPSOCSet,
+    MPStaticSet,
+)
+from pymatgen.io.vasp.sets.mvl import (
+    MVLElasticSet,
+    MVLGBSet,
+    MVLGWSet,
+    MVLNPTMDSet,
+    MVLRelax52Set,
+    MVLScanRelaxSet,
+    MVLSlabSet,
+)