Skip to content

Commit f8a0b31

Browse files
committed
[r2] fix NeighborStat (#207)
* fix NeighborStat Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu> * mixed_type Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu> --------- Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
1 parent 28c144a commit f8a0b31

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

deepmd/utils/neighbor_stat.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,9 @@ def __init__(
179179
one_type: bool = False,
180180
) -> None:
181181
"""Constructor."""
182-
super().__init__(ntypes, rcut, one_type)
182+
self.ntypes = ntypes
183+
self.rcut = rcut
184+
self.mixed_type = one_type
183185
self.auto_batch_size = AutoBatchSize()
184186
self.neighbor_stat = NeighborStatOP(ntypes, rcut, not one_type)
185187
self.place_holders = {}

0 commit comments

Comments
 (0)