style

PennyLaneAI · Aug 2, 2024 · 5657228 · 5657228
1 parent 4d7e9e3
commit 5657228
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/frontend/test/lit/test_skip_initial_state_prep.py b/frontend/test/lit/test_skip_initial_state_prep.py
@@ -47,15 +47,7 @@ def basis_state_example():
 
     Modified to use jax.numpy and a non trivial StatePrep
     """
-    qml.BasisState(
-        jnp.array(
-            [
-                1,
-                1,
-            ]
-        ),
-        wires=range(2),
-    )
+    qml.BasisState(jnp.array([1, 1]), wires=range(2))
     return qml.state()