try

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 9, 2024 · 4e42105 · 4e42105
1 parent b0a496c
commit 4e42105
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 4 deletions.
diff --git a/source/tests/jax/jax2tf/test_nlist.py b/source/tests/jax/jax2tf/test_nlist.py
@@ -1,6 +1,7 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 
-import pytest
+import unittest
+
 import tensorflow as tf
 import tensorflow.experimental.numpy as tnp
 
@@ -20,7 +21,7 @@
     dtype = tnp.float64
 
 
-@pytest.mark.skipif(
+@unittest.skipIf(
     not DP_TEST_TF2_ONLY,
     reason="TF2 conflicts with TF1",
 )

diff --git a/source/tests/jax/jax2tf/test_region.py b/source/tests/jax/jax2tf/test_region.py
@@ -1,7 +1,8 @@
 # SPDX-License-Identifier: LGPL-3.0-or-later
 
 
-import pytest
+import unittest
+
 import tensorflow as tf
 import tensorflow.experimental.numpy as tnp
 
@@ -19,7 +20,7 @@
     )
 
 
-@pytest.mark.skipif(
+@unittest.skipIf(
     not DP_TEST_TF2_ONLY,
     reason="TF2 conflicts with TF1",
 )