Fix scheduler unit test

Qiskit · Feb 20, 2024 · ea45b1b · ea45b1b
1 parent 4537952
commit ea45b1b
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/test/unit/transpiler/passes/scheduling/test_scheduler.py b/test/unit/transpiler/passes/scheduling/test_scheduler.py
@@ -20,10 +20,7 @@
 from qiskit.transpiler.passmanager import PassManager
 from qiskit.transpiler.exceptions import TranspilerError
 
-try:
-    from qiskit.providers.fake_provider import Fake7QPulseV1
-except ImportError:
-    from qiskit.providers.fake_provider import FakeJakarta as Fake7QPulseV1
+from qiskit_ibm_runtime.fake_provider import FakeJakarta
 
 from qiskit_ibm_provider.transpiler.passes.scheduling.pad_delay import PadDelay
 from qiskit_ibm_provider.transpiler.passes.scheduling.scheduler import (
@@ -850,12 +847,12 @@ def test_c_if_plugin_conversion_with_transpile(self):
         after transpilation with the plugin."""
         # Patch the test backend with the plugin
         with patch.object(
-            Fake7QPulseV1,
+            FakeJakarta,
             "get_translation_stage_plugin",
             return_value="ibm_dynamic_circuits",
             create=True,
         ):
-            backend = Fake7QPulseV1()
+            backend = FakeJakarta()
             # Temporary workaround for mock backends. For real backends this is not required.
             backend.configuration().basis_gates.append("if_else")
 
@@ -1867,7 +1864,7 @@ def test_for_loop(self):
 
     def test_transpile_mock_backend(self):
         """Test scheduling works with transpilation."""
-        backend = Fake7QPulseV1()
+        backend = FakeJakarta()
         # Temporary workaround for mock backends. For real backends this is not required.
         backend.configuration().basis_gates.append("if_else")
         backend.configuration().basis_gates.append("while_loop")
@@ -1915,7 +1912,7 @@ def test_transpile_mock_backend(self):
 
     def test_transpile_both_paths(self):
         """Test scheduling works with both fast- and standard path after transpiling."""
-        backend = Fake7QPulseV1()
+        backend = FakeJakarta()
         # Temporary workaround for mock backends. For real backends this is not required.
         backend.configuration().basis_gates.append("if_else")
 
@@ -1957,12 +1954,12 @@ def test_c_if_plugin_conversion_with_transpile(self):
         transpilation with the plugin."""
         # Patch the test backend with the plugin
         with patch.object(
-            Fake7QPulseV1,
+            FakeJakarta,
             "get_translation_stage_plugin",
             return_value="ibm_dynamic_circuits",
             create=True,
         ):
-            backend = Fake7QPulseV1()
+            backend = FakeJakarta()
             # Temporary workaround for mock backends. For real backends this is not required.
             backend.configuration().basis_gates.append("if_else")