minor cleanup

tests/test_eos.py

@@ -1,14 +1,19 @@
+
+import sys
+
 import pytest
 from ase.build import bulk
-import sys
 
 from mlip_arena.models.utils import MLIPEnum
 from mlip_arena.tasks.eos.run import fit as EOS
 
 atoms = bulk("Cu", "fcc", a=3.6)
 
+
 @pytest.mark.parametrize("model", [MLIPEnum["MACE-MP(M)"]])
-@pytest.mark.skipif(sys.version_info != (3, 11), reason="requires Python 3.11 to use prefect")
+@pytest.mark.skipif(
+    sys.version_info != (3, 11), reason="requires Python 3.11 to use prefect"
+)
 def test_eos(model: MLIPEnum):
     """
     Test EOS prefect workflow with a simple cubic lattice.

tests/test_external_calculators.py

@@ -5,7 +5,7 @@ from mlip_arena.models.utils import MLIPEnum
 
 
 @pytest.mark.parametrize("model", MLIPEnum)
-def test_calculate(model):
+def test_calculate(model: MLIPEnum):
     calc = MLIPEnum[model.name].value()
 
     atoms = Atoms(