Add UT for consistency

deepmodeling · Jun 3, 2024 · 9ae4dc7 · 9ae4dc7
1 parent c46fb27
commit 9ae4dc7
Show file tree

Hide file tree

Showing 5 changed files with 847 additions and 119 deletions.
diff --git a/deepmd/dpmodel/descriptor/dpa1.py b/deepmd/dpmodel/descriptor/dpa1.py
@@ -964,7 +964,7 @@ def serialize(self) -> dict:
         obj = self.se_atten
         data = {
             "@class": "Descriptor",
-            "type": "se_attn_v2",
+            "type": "se_atten_v2",
             "@version": 1,
             "rcut": obj.rcut,
             "rcut_smth": obj.rcut_smth,

diff --git a/deepmd/tf/descriptor/__init__.py b/deepmd/tf/descriptor/__init__.py
@@ -26,8 +26,6 @@
 )
 from .se_atten import (
     DescrptSeAtten,
-)
-from .se_atten_v2 import (
     DescrptSeAttenV2,
 )
 from .se_r import (