[r2] fix NeighborStat (#207)

* fix NeighborStat Signed-off-by: Jinzhe Zeng <[email protected]> * mixed_type Signed-off-by: Jinzhe Zeng <[email protected]> --------- Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Apr 6, 2024 · f8a0b31 · f8a0b31
1 parent 28c144a
commit f8a0b31
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/deepmd/utils/neighbor_stat.py b/deepmd/utils/neighbor_stat.py
@@ -179,7 +179,9 @@ def __init__(
         one_type: bool = False,
     ) -> None:
         """Constructor."""
-        super().__init__(ntypes, rcut, one_type)
+        self.ntypes = ntypes
+        self.rcut = rcut
+        self.mixed_type = one_type
         self.auto_batch_size = AutoBatchSize()
         self.neighbor_stat = NeighborStatOP(ntypes, rcut, not one_type)
         self.place_holders = {}