fix skip testing. I am still confused why it doesn't work

Signed-off-by: Jinzhe Zeng <[email protected]>
njzjz · Nov 7, 2024 · 5e10621 · 5e10621
1 parent 8bef185
commit 5e10621
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 8 deletions.
diff --git a/source/tests/jax/jax2tf/test_nlist.py b/source/tests/jax/jax2tf/test_nlist.py
@@ -20,11 +20,10 @@
     dtype = tnp.float64
 
 
-pytest.mark.skipif(
-    not DP_TEST_TF2_ONLY, reason="TF2 conflicts with TF1", allow_module_level=True
+@pytest.mark.skipif(
+    not DP_TEST_TF2_ONLY,
+    reason="TF2 conflicts with TF1",
 )
-
-
 class TestNeighList(tf.test.TestCase):
     def setUp(self):
         self.nf = 3

diff --git a/source/tests/jax/jax2tf/test_region.py b/source/tests/jax/jax2tf/test_region.py
@@ -18,11 +18,11 @@
         to_face_distance,
     )
 
-pytest.mark.skipif(
-    not DP_TEST_TF2_ONLY, reason="TF2 conflicts with TF1", allow_module_level=True
-)
-
 
+@pytest.mark.skipif(
+    not DP_TEST_TF2_ONLY,
+    reason="TF2 conflicts with TF1",
+)
 class TestRegion(tf.test.TestCase):
     def setUp(self):
         self.cell = tnp.array(