From adf8a73ef21044bd5a7fbca08a73595c1fba7351 Mon Sep 17 00:00:00 2001
From: Stavros Efthymiou <35475381+stavros11@users.noreply.github.com>
Date: Tue, 5 Dec 2023 15:55:47 +0400
Subject: [PATCH] style: simplify loop over readout pulse in sweep

---
 src/qibolab/instruments/zhinst.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/qibolab/instruments/zhinst.py b/src/qibolab/instruments/zhinst.py
index 758e6c4d28..ccd67167c7 100644
--- a/src/qibolab/instruments/zhinst.py
+++ b/src/qibolab/instruments/zhinst.py
@@ -1062,15 +1062,15 @@ def sweep(self, qubits, couplers, sequence: PulseSequence, options, *sweepers):
         for qubit in qubits.values():
             q = qubit.name  # pylint: disable=C0103
             if len(self.sequence[f"readout{q}"]) != 0:
-                for i in range(len(self.sequence[f"readout{q}"])):
+                for i, ropulse in enumerate(self.sequence[f"readout{q}"]):
                     exp_res = self.results.get_data(f"sequence{q}_{i}")
                     # Reorder dimensions
                     data = np.moveaxis(exp_res, rearranging_axes[0], rearranging_axes[1])
                     if options.acquisition_type is AcquisitionType.DISCRIMINATION:
                         data = np.ones(data.shape) - data.real  # Probability inversion patch
 
-                    serial = self.sequence[f"readout{q}"][i].pulse.serial
-                    qubit = self.sequence[f"readout{q}"][i].pulse.qubit
+                    serial = ropulse.pulse.serial
+                    qubit = ropulse.pulse.qubit
                     results[serial] = results[qubit] = options.results_type(data)
 
         self.offsets_off()