Linting

materialsproject · Jan 10, 2024 · 386875d · 386875d
1 parent 513c4e2
commit 386875d
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/emmet-builders/tests/test_ml.py b/emmet-builders/tests/test_ml.py
@@ -1,5 +1,3 @@
-from typing import Union
-
 import pytest
 from maggma.stores import MemoryStore
 
@@ -33,7 +31,7 @@ def materials_store():
 
 # @pytest.mark.parametrize("model", [get_universal_calculator("chgnet"), "m3gnet"])
 @pytest.mark.skip(reason="Temporary skip. Needs attention.")
-def test_ml_ip_builder(materials_store: MemoryStore, model: Union[str, "Calculator"]):
+def test_ml_ip_builder(materials_store: MemoryStore, model):
     ml_store = MemoryStore(key="material_id")
 
     builder = MLBuilder(materials=materials_store, ml_potential=ml_store, model=model)

diff --git a/emmet-core/tests/test_ml.py b/emmet-core/tests/test_ml.py
@@ -1,5 +1,3 @@
-from typing import Union
-
 import pytest
 
 # from matcalc.utils import get_universal_calculator
@@ -57,7 +55,7 @@
 #    ],
 # )
 @pytest.mark.skip(reason="Temporary skip. Needs attention.")
-def test_ml_doc(calculator: Union[str, "Calculator"], prop_kwargs: dict) -> None:
+def test_ml_doc(calculator, prop_kwargs: dict) -> None:
     doc = MLDoc(
         structure=struct,
         calculator=calculator,