Merge pull request #73 from Ericgig/bug.test_mc

Fix contradicting declaration
qutip · Dec 16, 2024 · c52d9e5 · c52d9e5
2 parents 4067dfa + 5550bf8
commit c52d9e5
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)