Update deepmd/jax/jax2tf/tfmodel.py

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Nov 3, 2024 · 980b4a9 · 980b4a9
1 parent 5a7fc4a
commit 980b4a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/jax/jax2tf/tfmodel.py b/deepmd/jax/jax2tf/tfmodel.py
@@ -266,7 +266,7 @@ def get_min_nbor_dist(self) -> Optional[float]:
 
     def get_nnei(self) -> int:
         """Returns the total number of selected neighboring atoms in the cut-off radius."""
-        return self.nsel
+        return self.get_nsel()
 
     def get_sel(self) -> list[int]:
         return self.sel