also add get_dim_fparam and get_dim_aparam

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Jan 24, 2024 · af6b05e · af6b05e
1 parent bbd6170
commit af6b05e
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/deepmd_utils/infer/deep_pot.py b/deepmd_utils/infer/deep_pot.py
@@ -130,5 +130,13 @@ def get_ntypes(self) -> int:
     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"]