Apply suggestions from code review

Co-authored-by: coderabbitai[bot] <136622811+coderabbitai[bot]@users.noreply.github.com> Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Oct 15, 2024 · 0d6d760 · 0d6d760
1 parent 9cf5143
commit 0d6d760
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/source/tests/consistent/descriptor/test_se_e2_a.py b/source/tests/consistent/descriptor/test_se_e2_a.py
@@ -44,7 +44,7 @@
         DescrptSeA as DescrptSeAArrayAPIStrict,
     )
 else:
-    DescrptSeAArrayAPI = None
+    DescrptSeAArrayAPIStrict = None
 
 
 @parameterized(
@@ -131,7 +131,7 @@ def skip_array_api_strict(self) -> bool:
             precision,
             env_protection,
         ) = self.param
-        return not type_one_side or not INSTALLED_JAX
+        return not type_one_side or not INSTALLED_ARRAY_API_STRICT
 
     tf_class = DescrptSeATF
     dp_class = DescrptSeADP