removed superfluous unused NeighborSpoof class

apax-hub · Nov 11, 2023 · 990fe82 · 990fe82
1 parent dea322f
commit 990fe82
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 18 deletions.
diff --git a/apax/model/gmnn.py b/apax/model/gmnn.py
@@ -16,7 +16,6 @@
 from apax.layers.properties import stress_times_vol
 from apax.layers.readout import AtomisticReadout
 from apax.layers.scaling import PerElementScaleShift
-from apax.model.utils import NeighborSpoof
 from apax.utils.math import fp64_sum
 
 DisplacementFn = Callable[[Array, Array], Array]
@@ -26,11 +25,7 @@
 
 
 def canonicalize_neighbors(neighbor):
-    return (
-        neighbor.idx
-        if isinstance(neighbor, (partition.NeighborList, NeighborSpoof))
-        else neighbor
-    )
+    return neighbor.idx if isinstance(neighbor, partition.NeighborList) else neighbor
 
 
 def disp_fn(ri, rj, perturbation, box):
@@ -96,7 +91,7 @@ def __call__(
         self,
         R: Array,
         Z: Array,
-        neighbor: Union[partition.NeighborList, NeighborSpoof, Array],
+        neighbor: Union[partition.NeighborList, Array],
         box,
         offsets,
         perturbation=None,
@@ -143,7 +138,7 @@ def __call__(
         self,
         R: Array,
         Z: Array,
-        neighbor: Union[partition.NeighborList, NeighborSpoof, Array],
+        neighbor: Union[partition.NeighborList, Array],
         box,
         offsets,
     ):

diff --git a/apax/model/utils.py b/apax/model/utils.py
diff --git a/tests/unit_tests/model/test_apax.py b/tests/unit_tests/model/test_apax.py
@@ -116,8 +116,6 @@ def test_energy_model():
         ]
     )
     offsets = jnp.full([6, 3], 0)
-    # neighbor = NeighborSpoof(idx=idx)
-
     box = np.array([0.0, 0.0, 0.0])
 
     model = EnergyModel()