Update source/tests/consistent/descriptor/test_se_r.py

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <jinzhe.zeng@rutgers.edu>
deepmodeling · Oct 26, 2024 · b8efe11 · b8efe11
1 parent 31c40b8
commit b8efe11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/tests/consistent/descriptor/test_se_r.py b/source/tests/consistent/descriptor/test_se_r.py
@@ -38,13 +38,13 @@
 if INSTALLED_JAX:
     from deepmd.jax.descriptor.se_e2_r import DescrptSeR as DescrptSeRJAX
 else:
-    DescrptSeAJAX = None
+    DescrptSeRJAX = None
 if INSTALLED_ARRAY_API_STRICT:
     from ...array_api_strict.descriptor.se_e2_r import (
         DescrptSeR as DescrptSeRArrayAPIStrict,
     )
 else:
-    DescrptSeAArrayAPIStrict = None
+    DescrptSeRArrayAPIStrict = None
 
 
 @parameterized(