From 5e1062157422a30dd46fe44ada3ad06461382492 Mon Sep 17 00:00:00 2001 From: Jinzhe Zeng Date: Thu, 7 Nov 2024 04:49:52 -0500 Subject: [PATCH] fix skip testing. I am still confused why it doesn't work Signed-off-by: Jinzhe Zeng --- source/tests/jax/jax2tf/test_nlist.py | 7 +++---- source/tests/jax/jax2tf/test_region.py | 8 ++++---- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/source/tests/jax/jax2tf/test_nlist.py b/source/tests/jax/jax2tf/test_nlist.py index 3b380bc2cc..049d948f84 100644 --- 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 index ce2536ddb0..f38b661401 100644 --- 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(