skip the whole module

Signed-off-by: Jinzhe Zeng <[email protected]>
deepmodeling · Nov 7, 2024 · a00aae8 · a00aae8
1 parent bd27d4f
commit a00aae8
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/source/tests/jax/jax2tf/__init__.py b/source/tests/jax/jax2tf/__init__.py
@@ -5,4 +5,6 @@
     DP_TEST_TF2_ONLY,
 )
 
-pytest.mark.skipif(not DP_TEST_TF2_ONLY, reason="TF2 conflicts with TF1")
+pytest.mark.skipif(
+    not DP_TEST_TF2_ONLY, reason="TF2 conflicts with TF1", allow_module_level=True
+)