Linting

materialsproject · Jan 10, 2024 · dc07696 · dc07696
1 parent 87355cb
commit dc07696
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 9 deletions.
diff --git a/emmet-builders/tests/test_ml.py b/emmet-builders/tests/test_ml.py
@@ -1,15 +1,15 @@
-from typing import TYPE_CHECKING, Union
+from typing import Union
 
 import pytest
 from maggma.stores import MemoryStore
-from matcalc.utils import get_universal_calculator
+
+# from matcalc.utils import get_universal_calculator
 from pymatgen.core import Lattice, Structure
 
 from emmet.builders.materials.ml import MLBuilder
 
-if TYPE_CHECKING:
-    from ase.calculators.calculator import Calculator
-
+# if TYPE_CHECKING:
+#    from ase.calculators.calculator import Calculator
 
 material_id = "1234"
 

diff --git a/emmet-core/tests/test_ml.py b/emmet-core/tests/test_ml.py
@@ -1,15 +1,16 @@
-from typing import TYPE_CHECKING, Union
+from typing import Union
 
 import pytest
-from matcalc.utils import get_universal_calculator
+
+# from matcalc.utils import get_universal_calculator
 from pymatgen.core import Structure
 from pymatgen.util.testing import PymatgenTest
 
 from emmet.core.elasticity import BulkModulus, ElasticTensorDoc, ShearModulus
 from emmet.core.ml import MLDoc
 
-if TYPE_CHECKING:
-    from ase.calculators.calculator import Calculator
+# if TYPE_CHECKING:
+#    from ase.calculators.calculator import Calculator
 
 struct = PymatgenTest.get_structure("Si")