feat: add serialize and deserialize to pt pairtab

deepmd/pt/model/model/pair_tab.py

@@ -90,6 +90,20 @@ def get_sel(self) -> int:
     def distinguish_types(self) -> bool:
         # to match DPA1 and DPA2.
         return False
+
+    def serialize(self) -> dict:
+        return {
+            "tab_file": self.tab_file,
+            "rcut": self.rcut,
+            "sel": self.sel
+        }
+
+    @classmethod
+    def deserialize(cls, data) -> "PairTabModel":
+        tab_file = data["tab_file"]
+        rcut = data["rcut"]
+        sel = data["sel"]
+        return cls(tab_file, rcut, sel)
 
     def forward_atomic(
         self,

source/tests/pt/test_pairtab.py

@@ -70,6 +70,20 @@ def test_with_mask(self):
 
     def test_jit(self):
         model = torch.jit.script(self.model)
+
+    @patch("numpy.loadtxt")
+    def test_deserialize(self, mock_loadtxt):
+        file_path = "dummy_path"
+        mock_loadtxt.return_value = np.array(
+            [
+                [0.005, 1.0, 2.0, 3.0],
+                [0.01, 0.8, 1.6, 2.4],
+                [0.015, 0.5, 1.0, 1.5],
+                [0.02, 0.25, 0.4, 0.75],
+            ]
+        )
+        model1 = PairTabModel.deserialize(self.model.serialize())
+        model1 = torch.jit.script(model1)
 
 
 class TestPairTabTwoAtoms(unittest.TestCase):