simplify init method of se_e2_a descriptor. fig bug in consistent UT

Author: Han Wang

deepmd/pt_expt/descriptor/se_e2_a.py

@@ -27,7 +27,6 @@ class DescrptSeA(DescrptSeADP, torch.nn.Module):
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         torch.nn.Module.__init__(self)
         DescrptSeADP.__init__(self, *args, **kwargs)
-        self._convert_state()
 
     def __setattr__(self, name: str, value: Any) -> None:
         if name in {"davg", "dstd"} and "_buffers" in self.__dict__:
@@ -53,30 +52,6 @@ def __setattr__(self, name: str, value: Any) -> None:
             return super().__setattr__(name, value)
         return super().__setattr__(name, value)
 
-    def _convert_state(self) -> None:
-        if self.davg is not None:
-            davg = torch.as_tensor(self.davg, device=env.DEVICE)
-            if "davg" in self._buffers:
-                self._buffers["davg"] = davg
-            else:
-                if hasattr(self, "davg"):
-                    delattr(self, "davg")
-                self.register_buffer("davg", davg)
-        if self.dstd is not None:
-            dstd = torch.as_tensor(self.dstd, device=env.DEVICE)
-            if "dstd" in self._buffers:
-                self._buffers["dstd"] = dstd
-            else:
-                if hasattr(self, "dstd"):
-                    delattr(self, "dstd")
-                self.register_buffer("dstd", dstd)
-        if self.embeddings is not None:
-            self.embeddings = NetworkCollection.deserialize(self.embeddings.serialize())
-        if self.emask is not None:
-            self.emask = PairExcludeMask(
-                self.ntypes, exclude_types=list(self.emask.get_exclude_types())
-            )
-
     def forward(
         self,
         nlist: torch.Tensor,

source/tests/consistent/common.py

@@ -92,7 +92,7 @@ class CommonTest(ABC):
     """Native DP model class."""
     pt_class: ClassVar[type | None]
     """PyTorch model class."""
-    pt_expt_class: ClassVar[type | None]
+    pt_expt_class: ClassVar[type | None] = None
     """PyTorch exportable model class."""
     jax_class: ClassVar[type | None]
     """JAX model class."""