fix shape jax random

pasqal-io · Mar 7, 2025 · 1bbc662 · 1bbc662
1 parent f755efa
commit 1bbc662
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_differentiation.py b/tests/test_differentiation.py
@@ -27,7 +27,7 @@
 def test_shots(same_name: bool) -> None:
     param_name = "theta"
     if same_name:
-        x = jax.random.uniform(jax.random.key(0), (1))
+        x = jax.random.uniform(jax.random.key(0), 1)
         ops = [RX(param_name, 0), RX(param_name, 1)]
 
         def values_to_dict(x):