update doc

deepmodeling · Jan 27, 2024 · 4cc9f70 · 4cc9f70
1 parent eff9f69
commit 4cc9f70
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/deepmd/model_format/se_e2_a.py b/deepmd/model_format/se_e2_a.py
@@ -227,11 +227,13 @@ def call(
         gr
             The rotationally equivariant and permutationally invariant single particle
             representation. shape: nf x nloc x ng x 3
-        gg
-            The rotationally invariant pair-partical channel, this descriptor returns None
-        rr
-            The rotationally equivariant pair-partical channel, this descriptor returns None
-        ww
+        g2
+            The rotationally invariant pair-partical representation.
+            this descriptor returns None
+        h2
+            The rotationally equivariant pair-partical representation.
+            this descriptor returns None
+        sw
             The smooth switch function.
         """
         # nf x nloc x nnei x 4