chore: halting test/test_optimizers::test_numpyrooptim_no_double_jit till pyro-pplgh-2137 is merged

Qazalbash · Qazalbash · commit d394188f268a · 2026-02-12T14:55:36.000+05:00
diff --git a/test/test_optimizers.py b/test/test_optimizers.py
@@ -100,6 +100,7 @@ def test_optim_multi_params(optim_class, args, kwargs, uses_value_arg):
     + optax_optimizers,
 )
 @pytest.mark.filterwarnings("ignore:.*tree_multimap:FutureWarning")
+@pytest.mark.skip(reason="Failing on optax==0.2.7. Halting till #2137 is merged.")
 def test_numpyrooptim_no_double_jit(optim_class, args, kwargs, uses_value_arg):
     opt = optim_class(*args, **kwargs)
     if not isinstance(opt, optim._NumPyroOptim):

Original file line number	Diff line number	Diff line change
`@@ -100,6 +100,7 @@ def test_optim_multi_params(optim_class, args, kwargs, uses_value_arg):`
`100`	`100`	`+ optax_optimizers,`
`101`	`101`	`)`
`102`	`102`	`@pytest.mark.filterwarnings("ignore:.*tree_multimap:FutureWarning")`
	`103`	`+@pytest.mark.skip(reason="Failing on optax==0.2.7. Halting till #2137 is merged.")`
`103`	`104`	`def test_numpyrooptim_no_double_jit(optim_class, args, kwargs, uses_value_arg):`
`104`	`105`	`opt = optim_class(args, *kwargs)`
`105`	`106`	`if not isinstance(opt, optim._NumPyroOptim):`