Merge branch 'add_dpa2_v22' into add_dpa2_v2

deepmodeling · Dec 2, 2024 · 984b242 · 984b242
2 parents 89d82f9 + ed80c6d
commit 984b242
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepmd/pd/model/descriptor/__init__.py b/deepmd/pd/model/descriptor/__init__.py
@@ -26,11 +26,11 @@
 __all__ = [
     "BaseDescriptor",
     "DescriptorBlock",
+    "DescrptBlockRepformers",
     "DescrptBlockSeA",
     "DescrptBlockSeAtten",
     "DescrptDPA1",
     "DescrptDPA2",
     "DescrptSeA",
     "prod_env_mat",
-    "DescrptBlockRepformers",
 ]