Add more type-hints

src/quacc/recipes/emt/defects.py

@@ -81,7 +81,7 @@
     if "relax_cell" not in defect_relax_kwargs:
         defect_relax_kwargs["relax_cell"] = False
 
-    def _make_defects(atoms):
+    def _make_defects(atoms: Atoms) -> list[Atoms]:
         return make_defects_from_bulk(
             atoms,
             defect_gen=defect_gen,
@@ -90,12 +90,12 @@
         )
 
     @subflow
-    def _relax_distributed(atoms):
+    def _relax_distributed(atoms: Atoms) -> list[OptSchema]:
         defects = _make_defects(atoms)
         return [relax_job(defect, **defect_relax_kwargs) for defect in defects]
 
     @subflow
-    def _relax_and_static_distributed(atoms):
+    def _relax_and_static_distributed(atoms: Atoms) -> list[RunSchema]:
         defects = _make_defects(atoms)
         return [
             static_job(

src/quacc/recipes/emt/slabs.py

@@ -59,16 +59,16 @@ def bulk_to_slabs_flow(
     if "relax_cell" not in slab_relax_kwargs:
         slab_relax_kwargs["relax_cell"] = False
 
-    def _make_slabs(atoms):
+    def _make_slabs(atoms: Atoms) -> list[Atoms]:
         return make_slabs_from_bulk(atoms, **make_slabs_kwargs)
 
     @subflow
-    def _relax_distributed(atoms):
+    def _relax_distributed(atoms: Atoms) -> list[OptSchema]:
         slabs = _make_slabs(atoms)
         return [relax_job(slab, **slab_relax_kwargs) for slab in slabs]
 
     @subflow
-    def _relax_and_static_distributed(atoms):
+    def _relax_and_static_distributed(atoms: Atoms) -> list[OptSchema]:
         slabs = _make_slabs(atoms)
         return [
             static_job(

src/quacc/recipes/vasp/slabs.py

@@ -186,16 +186,16 @@
     slab_static_kwargs = slab_static_kwargs or {}
     make_slabs_kwargs = make_slabs_kwargs or {}
 
-    def _make_slabs(atoms):
+    def _make_slabs(atoms: Atoms) -> list[Atoms]:
         return make_slabs_from_bulk(atoms, **make_slabs_kwargs)
 
     @subflow
-    def _relax_distributed(atoms):
+    def _relax_distributed(atoms: Atoms) -> list[VaspSchema]:
         slabs = _make_slabs(atoms)
         return [slab_relax_job(slab, **slab_relax_kwargs) for slab in slabs]
 
     @subflow
-    def _relax_and_static_distributed(atoms):
+    def _relax_and_static_distributed(atoms: Atoms) -> list[VaspSchema]:
         slabs = _make_slabs(atoms)
         return [
             slab_static_job(
@@ -255,16 +255,16 @@
     slab_static_kwargs = slab_static_kwargs or {}
     make_ads_kwargs = make_ads_kwargs or {}
 
-    def _make_ads_slabs(slab):
+    def _make_ads_slabs(slab: Atoms) -> list[Atoms]:
         return make_adsorbate_structures(slab, adsorbate, **make_ads_kwargs)
 
     @subflow
-    def _relax_distributed(slab):
+    def _relax_distributed(slab: Atoms) -> list[VaspSchema]:
         slabs = _make_ads_slabs(slab)
         return [slab_relax_job(slab, **slab_relax_kwargs) for slab in slabs]
 
     @subflow
-    def _relax_and_static_distributed(slab):
+    def _relax_and_static_distributed(slab: Atoms) -> list[VaspSchema]:
         slabs = _make_ads_slabs(slab)
         return [
             slab_static_job(