feat: add empty compress method to Pairtab

deepmodeling · Nov 27, 2024 · 8b94a81 · 8b94a81
1 parent f343a3b
commit 8b94a81
Show file tree

Hide file tree

Showing 2 changed files with 22 additions and 0 deletions.
diff --git a/deepmd/dpmodel/atomic_model/linear_atomic_model.py b/deepmd/dpmodel/atomic_model/linear_atomic_model.py
@@ -427,6 +427,17 @@ def deserialize(cls, data) -> "DPZBLLinearEnergyAtomicModel":
         data.pop("@class", None)
         data.pop("type", None)
         return super().deserialize(data)
+
+    def enable_compression(
+        self,
+        min_nbor_dist: float,
+        table_extrapolate: float = 5,
+        table_stride_1: float = 0.01,
+        table_stride_2: float = 0.1,
+        check_frequency: int = -1,
+    ) -> None:
+        """Pairtab model does not support compression."""
+        pass
 
     def _compute_weight(
         self,

diff --git a/deepmd/pt/model/atomic_model/linear_atomic_model.py b/deepmd/pt/model/atomic_model/linear_atomic_model.py
@@ -573,6 +573,17 @@ def deserialize(cls, data) -> "DPZBLLinearEnergyAtomicModel":
         data.pop("@class", None)
         data.pop("type", None)
         return super().deserialize(data)
+
+    def enable_compression(
+        self,
+        min_nbor_dist: float,
+        table_extrapolate: float = 5,
+        table_stride_1: float = 0.01,
+        table_stride_2: float = 0.1,
+        check_frequency: int = -1,
+    ) -> None:
+        """Pairtab model does not support compression."""
+        pass
 
     def _compute_weight(
         self,