Fix contradicting declaration

qutip · Dec 16, 2024 · 5550bf8 · 5550bf8
1 parent 4067dfa
commit 5550bf8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_qutip/test_mcsolve.py b/tests/test_qutip/test_mcsolve.py
@@ -10,7 +10,7 @@
 qjax.set_as_default()
 
 # Define time-dependent functions
-@partial(jax.jit, static_argnames=("omega",))
+@jax.jit
 def H_1_coeff(t, omega):
     return 2.0 * jnp.pi * 0.25 * jnp.cos(2.0 * omega * t)