Fix type hints (#1179)

Quantum-Accelerators · Nov 9, 2023 · 734846c · 734846c
1 parent 761f0bd
commit 734846c
Show file tree

Hide file tree

Showing 18 changed files with 19 additions and 19 deletions.
diff --git a/src/quacc/recipes/common/slabs.py b/src/quacc/recipes/common/slabs.py
@@ -11,7 +11,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema, RunSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema
 
 
 def common_bulk_to_slabs_flow(

diff --git a/src/quacc/recipes/dftb/core.py b/src/quacc/recipes/dftb/core.py
@@ -16,7 +16,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import RunSchema
+ from quacc.schemas._aliases.ase import RunSchema
 
 LOG_FILE = "dftb.out"
 GEOM_FILE = "geo_end.gen"

diff --git a/src/quacc/recipes/emt/core.py b/src/quacc/recipes/emt/core.py
@@ -20,7 +20,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema, RunSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema
 
 
 @job

diff --git a/src/quacc/recipes/emt/defects.py b/src/quacc/recipes/emt/defects.py
@@ -21,7 +21,7 @@
  VoronoiInterstitialGenerator,
  )
 
- from quacc.schemas.ase import OptSchema, RunSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema
 
 
 @flow

diff --git a/src/quacc/recipes/emt/slabs.py b/src/quacc/recipes/emt/slabs.py
@@ -12,7 +12,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema, RunSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema
 
 
 @flow

diff --git a/src/quacc/recipes/gaussian/core.py b/src/quacc/recipes/gaussian/core.py
@@ -16,7 +16,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.cclib import cclibSchema
+ from quacc.schemas._aliases.cclib import cclibSchema
 
 _LABEL = "Gaussian"
 LOG_FILE = f"{_LABEL}.log"

diff --git a/src/quacc/recipes/gulp/core.py b/src/quacc/recipes/gulp/core.py
@@ -17,7 +17,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import RunSchema
+ from quacc.schemas._aliases.ase import RunSchema
 
 logger = logging.getLogger(__name__)
 

diff --git a/src/quacc/recipes/lj/core.py b/src/quacc/recipes/lj/core.py
@@ -22,7 +22,7 @@
  from ase import Atoms
 
  from quacc.runners.ase import VibKwargs
- from quacc.schemas.ase import OptSchema, RunSchema, VibThermoSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema, VibThermoSchema
 
 
 @job

diff --git a/src/quacc/recipes/newtonnet/core.py b/src/quacc/recipes/newtonnet/core.py
@@ -33,7 +33,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema, RunSchema, ThermoSchema, VibSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema, ThermoSchema, VibSchema
 
  class FreqSchema(RunSchema):
  vib: VibSchema

diff --git a/src/quacc/recipes/newtonnet/ts.py b/src/quacc/recipes/newtonnet/ts.py
@@ -28,7 +28,7 @@
  from numpy.typing import NDArray
 
  from quacc.recipes.newtonnet.core import FreqSchema
- from quacc.schemas.ase import OptSchema
+ from quacc.schemas._aliases.ase import OptSchema
 
  class TSSchema(OptSchema):
  freq_job: FreqSchema | None

diff --git a/src/quacc/recipes/orca/core.py b/src/quacc/recipes/orca/core.py
@@ -17,7 +17,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.cclib import cclibSchema
+ from quacc.schemas._aliases.cclib import cclibSchema
 
 _LABEL = OrcaTemplate()._label # skipcq: PYL-W0212
 LOG_FILE = f"{_LABEL}.out"

diff --git a/src/quacc/recipes/psi4/core.py b/src/quacc/recipes/psi4/core.py
@@ -21,7 +21,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import RunSchema
+ from quacc.schemas._aliases.ase import RunSchema
 
 
 @job

diff --git a/src/quacc/recipes/qchem/core.py b/src/quacc/recipes/qchem/core.py
@@ -24,7 +24,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema, RunSchema
+ from quacc.schemas._aliases.ase import OptSchema, RunSchema
 
 
 @job

diff --git a/src/quacc/recipes/qchem/ts.py b/src/quacc/recipes/qchem/ts.py
@@ -22,7 +22,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema
+ from quacc.schemas._aliases.ase import OptSchema
 
 
 @job

diff --git a/src/quacc/recipes/vasp/core.py b/src/quacc/recipes/vasp/core.py
@@ -14,7 +14,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.vasp import VaspSchema
+ from quacc.schemas._aliases.vasp import VaspSchema
 
  class DoubleRelaxSchema(VaspSchema):
  relax1: VaspSchema

diff --git a/src/quacc/recipes/vasp/mp.py b/src/quacc/recipes/vasp/mp.py
@@ -27,7 +27,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.vasp import VaspSchema
+ from quacc.schemas._aliases.vasp import VaspSchema
 
  class MPRelaxFlowSchema(VaspSchema):
  prerelax: VaspSchema

diff --git a/src/quacc/recipes/vasp/qmof.py b/src/quacc/recipes/vasp/qmof.py
@@ -20,8 +20,8 @@
 if TYPE_CHECKING:
  from ase import Atoms
 
- from quacc.schemas.ase import OptSchema
- from quacc.schemas.vasp import VaspSchema
+ from quacc.schemas._aliases.ase import OptSchema
+ from quacc.schemas._aliases.vasp import VaspSchema
 
  class QMOFRelaxSchema(VaspSchema):
  prerelax_lowacc: VaspSchema | None

diff --git a/src/quacc/recipes/vasp/slabs.py b/src/quacc/recipes/vasp/slabs.py
@@ -13,7 +13,7 @@
 
  from ase import Atoms
 
- from quacc.schemas.vasp import VaspSchema
+ from quacc.schemas._aliases.vasp import VaspSchema
 
 
 @job