dev(hansbug): fix skip

opendilab · Feb 26, 2023 · ee3184d · ee3184d
1 parent b6192f4
commit ee3184d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/tree/integration/test_jax.py b/test/tree/integration/test_jax.py
@@ -12,8 +12,8 @@
 
 
 @pytest.mark.unittest
-@skipUnless(jax, 'Jax required.')
 class TestTreeTreeIntegration:
+    @skipUnless(jax, 'Jax required.')
     def test_jax_double(self):
         @jax.jit
         def double(x):