diff --git a/deepmd_pt/infer/deep_eval.py b/deepmd_pt/infer/deep_eval.py index 3bc884e..790efa6 100644 --- a/deepmd_pt/infer/deep_eval.py +++ b/deepmd_pt/infer/deep_eval.py @@ -141,9 +141,9 @@ def _eval_model( atomic: bool = False, ): model = self.dp.to(DEVICE) - return eval(model, coords, cells, atom_types, atomic=atomic) + return eval_model(model, coords, cells, atom_types, atomic=atomic) -def eval( +def eval_model( model, coords: np.ndarray, cells: Optional[np.ndarray],