commit

PennyLaneAI · Jul 25, 2024 · f2f3efc · f2f3efc
1 parent cc3e25b
commit f2f3efc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/capture/test_templates.py b/tests/capture/test_templates.py
@@ -116,9 +116,9 @@ def enable_disable_plxpr():
  (qml.ArbitraryStatePreparation, (jnp.ones(6), [2, 3]), {}),
  (qml.ArbitraryStatePreparation, (jnp.zeros(14),), {"wires": [3, 2, 0]}),
  (qml.ArbitraryStatePreparation, (), {"weights": jnp.ones(2), "wires": [1]}),
- #(qml.BasisStatePreparation, (jnp.array([0, 1]), [2, 3]), {}),
- #(qml.BasisStatePreparation, (jnp.ones(3),), {"wires": [3, 2, 0]}),
- #(qml.BasisStatePreparation, (), {"basis_state": jnp.ones(1), "wires": [1]}),
+ (qml.BasisStatePreparation, (jnp.array([0, 1]), [2, 3]), {}),
+ (qml.BasisStatePreparation, (jnp.ones(3),), {"wires": [3, 2, 0]}),
+ (qml.BasisStatePreparation, (), {"basis_state": jnp.ones(1), "wires": [1]}),
  (qml.CosineWindow, ([2, 3],), {}),
  (qml.CosineWindow, (), {"wires": [2, 0, 1]}),
  (qml.MottonenStatePreparation, (jnp.ones(4) / 2, [2, 3]), {}),

diff --git a/tests/ops/qubit/test_state_prep.py b/tests/ops/qubit/test_state_prep.py
@@ -38,7 +38,7 @@ def test_adjoint_error_exception(op):
 def test_BasisStatePreparation_is_deprecated():
  """Test that my_feature is deprecated."""
  with pytest.warns(qml.PennyLaneDeprecationWarning, match="BasisStatePreparation is deprecated"):
- _ = qml.BasisStatePreparation()
+ _ = qml.BasisStatePreparation([1,0], wires = [0,1])
 
 @pytest.mark.parametrize(
  "op, mat, base",