Fix failing tests

PennyLaneAI · Jan 9, 2025 · 2f0bee9 · 2f0bee9
1 parent 8d88191
commit 2f0bee9
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/data/attributes/operator/test__wires.py b/tests/data/attributes/operator/test__wires.py
@@ -59,7 +59,7 @@ def test_wires_output_is_expected(self, in_, out):
             (0, 1, 2),
             range(3),
             ["a", "b"],
-            ["a", 0, 1, None],
+            ["a", 0, 1, "b"],
             1,
             "a",
             np.int64(3),

diff --git a/tests/pytrees/test_serialization.py b/tests/pytrees/test_serialization.py
@@ -97,7 +97,7 @@ def test_pytree_structure_dump(decode):
             "list": ["a", 1],
             "dict": {"a": 1},
             "tuple": ("a", 1),
-            "custom": CustomNode([1, 5, 7], {"wires": Wires([1, "a", 3.4, None])}),
+            "custom": CustomNode([1, 5, 7], {"wires": Wires([1, "a", 3.4, 2])}),
         }
     )
 
@@ -114,7 +114,7 @@ def test_pytree_structure_dump(decode):
                 [None],
             ],
             ["builtins.tuple", None, [None, None]],
-            ["test.CustomNode", {"wires": [1, "a", 3.4, None]}, [None, None, None]],
+            ["test.CustomNode", {"wires": [1, "a", 3.4, 2]}, [None, None, None]],
         ],
     ]