Skip to content

Commit

Permalink
rename
Browse files Browse the repository at this point in the history
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz committed Jan 18, 2024
1 parent 7d3f886 commit c5ead9e
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions deepmd_pt/infer/deep_eval.py
Original file line number Diff line number Diff line change
Expand Up @@ -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],
Expand Down

0 comments on commit c5ead9e

Please sign in to comment.