Skip to content

Commit d3a5a76

Browse files
committed
fix(pt): make int rcut safe after jit op
1 parent cfb4731 commit d3a5a76

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

deepmd/pt/utils/neighbor_stat.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def __init__(
4444
mixed_types: bool,
4545
) -> None:
4646
super().__init__()
47-
self.rcut = rcut
47+
self.rcut = float(rcut)
4848
self.ntypes = ntypes
4949
self.mixed_types = mixed_types
5050

0 commit comments

Comments
 (0)