Fix broken test and linting problems.

qua-platform · Oct 22, 2024 · 15ae400 · 15ae400
1 parent 7f9b6ed
commit 15ae400
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 11 deletions.
diff --git a/qualang_tools/callable_from_qua/_callable_from_qua.py b/qualang_tools/callable_from_qua/_callable_from_qua.py
@@ -79,13 +79,16 @@ def run(self, job: QmJob):
 
 class ProgramAddon(ABC):
     @abstractmethod
-    def enter_program(self, program: Program): ...  # noqa: E704
+    def enter_program(self, program: Program):
+        ...  # noqa: E704
 
     @abstractmethod
-    def exit_program(self, exc_type, exc_val, exc_tb): ...  # noqa: E704
+    def exit_program(self, exc_type, exc_val, exc_tb):
+        ...  # noqa: E704
 
     @abstractmethod
-    def execute_program(self, program: Program, quantum_machine: QuantumMachine): ...  # noqa: E704
+    def execute_program(self, program: Program, quantum_machine: QuantumMachine):
+        ...  # noqa: E704
 
 
 class QuaCallableEventManager(ProgramAddon):

diff --git a/qualang_tools/wirer/connectivity/wiring_spec.py b/qualang_tools/wirer/connectivity/wiring_spec.py
@@ -1,7 +1,5 @@
 from enum import Enum
 from typing import Union, List
-from .channel_spec import ChannelSpec
-from .element import Element
 
 from typing import TYPE_CHECKING
 

diff --git a/tests/wirer/test_wirer_channel_reuse.py b/tests/wirer/test_wirer_channel_reuse.py
@@ -57,23 +57,23 @@ def test_5q_allocation_with_channel_reuse(instruments_2lf_2mw):
             )
 
 
-def test_alternating_blocking_of_used_channels(instruments_1opx_1octave):
+def test_alternating_blocking_of_used_channels(instruments_2lf_2mw):
     connectivity = Connectivity()
 
     connectivity.add_qubit_drive_lines(qubits=1)
-    allocate_wiring(connectivity, instruments_1opx_1octave, block_used_channels=False)
+    allocate_wiring(connectivity, instruments_2lf_2mw, block_used_channels=False)
 
     connectivity.add_qubit_drive_lines(qubits=2)
-    allocate_wiring(connectivity, instruments_1opx_1octave)
+    allocate_wiring(connectivity, instruments_2lf_2mw)
 
     connectivity.add_qubit_drive_lines(qubits=3)
-    allocate_wiring(connectivity, instruments_1opx_1octave, block_used_channels=False)
+    allocate_wiring(connectivity, instruments_2lf_2mw, block_used_channels=False)
 
     connectivity.add_qubit_drive_lines(qubits=4)
-    allocate_wiring(connectivity, instruments_1opx_1octave)
+    allocate_wiring(connectivity, instruments_2lf_2mw)
 
     if visualize_flag:
-        visualize(connectivity.elements, instruments_1opx_1octave.available_channels)
+        visualize(connectivity.elements, instruments_2lf_2mw.available_channels)
 
     expected_ports = [
         1,  # q1 allocated to 1, but channel isn't blocked