fix assertion

rigetti · Aug 13, 2024 · c2d517b · c2d517b
1 parent ccf0dac
commit c2d517b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/unit/test_quilbase.py b/test/unit/test_quilbase.py
@@ -830,7 +830,7 @@ def test_convert(self, reset_qubit: Reset):
     def test_pickle(self, reset_qubit: Reset):
         pickled = pickle.dumps(reset_qubit)
         unpickled = pickle.loads(pickled)
-        assert isinstance(unpickled, Reset)
+        assert isinstance(unpickled, (Reset, ResetQubit))
         assert unpickled == reset_qubit