fix: use DEVICE.type in compile args

Author: CompRhys

tests/models/test_nequip_framework.py

@@ -39,7 +39,7 @@ def compiled_nequip_model_path(tmp_path_factory: pytest.TempPathFactory) -> Path
             "--mode",
             "aotinductor",
             "--device",
-            "cuda",
+            DEVICE.type,
             "--target",
             "ase",
         ]
@@ -48,7 +48,7 @@ def compiled_nequip_model_path(tmp_path_factory: pytest.TempPathFactory) -> Path
     return output_path
 
 
-@pytest.fixture
+@pytest.fixture(scope="session")
 def nequip_model(compiled_nequip_model_path: Path) -> NequIPFrameworkModel:
     """Create an NequIPModel wrapper for the pretrained model."""
     compiled_model, (r_max, type_names) = from_compiled_model(
@@ -62,16 +62,18 @@ def nequip_model(compiled_nequip_model_path: Path) -> NequIPFrameworkModel:
     )
 
 
-@pytest.fixture
-def nequip_calculator(model_path_nequip: Path) -> NequIPCalculator:
+@pytest.fixture(scope="session")
+def nequip_calculator(compiled_nequip_model_path: Path) -> NequIPCalculator:
     """Create an NequIPCalculator for the pretrained model."""
-    return NequIPCalculator.from_compiled_model(str(model_path_nequip), device=DEVICE)
+    return NequIPCalculator.from_compiled_model(
+        str(compiled_nequip_model_path), device=DEVICE
+    )
 
 
-def test_nequip_initialization(model_path_nequip: Path) -> None:
+def test_nequip_initialization(compiled_nequip_model_path: Path) -> None:
     """Test that the NequIP model initializes correctly."""
     compiled_model, (r_max, type_names) = from_compiled_model(
-        model_path_nequip, device=DEVICE
+        compiled_nequip_model_path, device=DEVICE
     )
     model = NequIPFrameworkModel(
         model=compiled_model,
@@ -82,7 +84,7 @@ def test_nequip_initialization(model_path_nequip: Path) -> None:
     assert model._device == DEVICE  # noqa: SLF001
 
 
-test_metatomic_consistency = make_model_calculator_consistency_test(
+test_nequip_consistency = make_model_calculator_consistency_test(
     test_name="nequip",
     model_fixture_name="nequip_model",
     calculator_fixture_name="nequip_calculator",
@@ -92,7 +94,7 @@ def test_nequip_initialization(model_path_nequip: Path) -> None:
     device=DEVICE,
 )
 
-test_metatomic_model_outputs = make_validate_model_outputs_test(
+test_nequip_model_outputs = make_validate_model_outputs_test(
     model_fixture_name="nequip_model",
     dtype=DTYPE,
     device=DEVICE,