Update to use DoWhileController

garrison · Mar 19, 2024 · ed91caa · ed91caa
1 parent d88605c
commit ed91caa
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/python/transpiler/test_remove_final_reset.py b/test/python/transpiler/test_remove_final_reset.py
@@ -15,6 +15,7 @@
 import unittest
 
 from qiskit import QuantumRegister, QuantumCircuit
+from qiskit.passmanager.flow_controllers import DoWhileController
 from qiskit.transpiler import PassManager
 from qiskit.transpiler.passes import RemoveFinalReset, DAGFixedPoint
 from qiskit.converters import circuit_to_dag
@@ -102,8 +103,10 @@ def test_two_resets(self):
 
         pass_manager = PassManager()
         pass_manager.append(
-            [RemoveFinalReset(), DAGFixedPoint()],
-            do_while=lambda property_set: not property_set["dag_fixed_point"],
+            DoWhileController(
+                [RemoveFinalReset(), DAGFixedPoint()],
+                do_while=lambda property_set: not property_set["dag_fixed_point"],
+            )
         )
         after = pass_manager.run(circuit)