1 parent 865f0d4 commit 7f2a952Copy full SHA for 7f2a952
1 file changed
deepmd/calculator.py
@@ -91,13 +91,15 @@ def __init__(
91
type_dict: dict[str, int] | None = None,
92
neighbor_list: Optional["NeighborList"] = None,
93
head: str | None = None,
94
+ no_jit: bool = False,
95
**kwargs: Any,
96
) -> None:
97
Calculator.__init__(self, label=label, **kwargs)
98
self.dp = DeepPot(
99
str(Path(model).resolve()),
100
neighbor_list=neighbor_list,
101
head=head,
102
+ no_jit=no_jit,
103
)
104
if type_dict:
105
self.type_dict = type_dict
0 commit comments