more linting

lab-cosmo · Jul 10, 2024 · a237de3 · a237de3
1 parent 19134e5
commit a237de3
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 105 deletions.
diff --git a/tests/metatensor/test_base_metatensor.py b/tests/metatensor/test_base_metatensor.py
@@ -2,13 +2,14 @@
 import torch
 from packaging import version
 
+import meshlode
+
 
 mts_torch = pytest.importorskip("metatensor.torch")
 mts_atomistic = pytest.importorskip("metatensor.torch.atomistic")
-meshlode_metatensor = pytest.importorskip("meshlode.metatensor")
 
 
-class CalculatorTest(meshlode_metatensor.base.CalculatorBaseMetatensor):
+class CalculatorTest(meshlode.metatensor.base.CalculatorBaseMetatensor):
     def _compute_single_system(
         self, positions, charges, cell, neighbor_indices, neighbor_shifts
     ):

diff --git a/tests/metatensor/test_calculators.py b/tests/metatensor/test_calculators.py
diff --git a/tests/metatensor/test_workflow_metatensor.py b/tests/metatensor/test_workflow_metatensor.py
@@ -6,10 +6,11 @@
 import torch
 from packaging import version
 
+import meshlode
+
 
 mts_torch = pytest.importorskip("metatensor.torch")
 mts_atomistic = pytest.importorskip("metatensor.torch.atomistic")
-meshlode_metatensor = pytest.importorskip("meshlode.metatensor")
 
 
 ATOMIC_SMEARING = 0.1
@@ -22,17 +23,17 @@
 @pytest.mark.parametrize(
     "CalculatorClass, params",
     [
-        (meshlode_metatensor.DirectPotential, {}),
+        (meshlode.metatensor.DirectPotential, {}),
         (
-            meshlode_metatensor.EwaldPotential,
+            meshlode.metatensor.EwaldPotential,
             {
                 "atomic_smearing": ATOMIC_SMEARING,
                 "lr_wavelength": LR_WAVELENGTH,
                 "subtract_self": SUBTRACT_SELF,
             },
         ),
         (
-            meshlode_metatensor.PMEPotential,
+            meshlode.metatensor.PMEPotential,
             {
                 "atomic_smearing": ATOMIC_SMEARING,
                 "mesh_spacing": MESH_SPACING,