Skip to content

Commit

Permalink
Apply suggestions from code review
Browse files Browse the repository at this point in the history
Co-authored-by: Han Wang <[email protected]>
Signed-off-by: Jinzhe Zeng <[email protected]>
  • Loading branch information
njzjz and wanghan-iapcm authored Feb 25, 2024
1 parent c2fb0be commit 7af6ce3
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions deepmd/pt/model/model/model.py
Original file line number Diff line number Diff line change
Expand Up @@ -156,15 +156,15 @@ def compute_or_load_stat(
@abstractmethod
def get_model_def_script(self) -> str:
"""Get the model definition script."""
raise NotImplementedError
pass

@abstractmethod
def get_nnei(self) -> int:
"""Returns the total number of selected neighboring atoms in the cut-off radius."""
# for C++ interface
raise NotImplementedError
pass

@abstractmethod
def get_nsel(self) -> int:
"""Returns the total number of selected neighboring atoms in the cut-off radius."""
raise NotImplementedError
pass

0 comments on commit 7af6ce3

Please sign in to comment.