diff --git a/deepmd_utils/infer/deep_pot.py b/deepmd_utils/infer/deep_pot.py index dec0a7c47c..66510ea349 100644 --- a/deepmd_utils/infer/deep_pot.py +++ b/deepmd_utils/infer/deep_pot.py @@ -122,5 +122,21 @@ def eval( # dpdata # ase + @abstractmethod + def get_ntypes(self) -> int: + """Get the number of atom types of this model.""" + + @abstractmethod + def get_type_map(self) -> List[str]: + """Get the type map (element name of the atom types) of this model.""" + + @abstractmethod + def get_dim_fparam(self) -> int: + """Get the number (dimension) of frame parameters of this DP.""" + + @abstractmethod + def get_dim_aparam(self) -> int: + """Get the number (dimension) of atomic parameters of this DP.""" + __all__ = ["DeepPot"]