Revert "fix: Missing qubit key because of identation"

This reverts commit 4ea3394.
qiboteam · Jul 4, 2024 · 5aa50ee · 5aa50ee
1 parent d340fd7
commit 5aa50ee
Showing 1 changed file with 35 additions and 35 deletions.
diff --git a/src/qibocal/protocols/readout_optimization/twpa_calibration/frequency_power.py b/src/qibocal/protocols/readout_optimization/twpa_calibration/frequency_power.py
@@ -100,46 +100,46 @@ def _acquisition(
         ].twpa.local_oscillator.frequency
         initial_twpa_power[qubit] = platform.qubits[qubit].twpa.local_oscillator.power
 
-    for freq in freq_range:
-        platform.qubits[qubit].twpa.local_oscillator.frequency = (
-            initial_twpa_freq[qubit] + freq
-        )
-
-        for power in power_range:
-            for qubit in targets:
-                platform.qubits[qubit].twpa.local_oscillator.power = (
-                    initial_twpa_power[qubit] + power
-                )
-
-            classification_data = classification._acquisition(
-                classification.SingleShotClassificationParameters.load(
-                    {"nshots": params.nshots}
-                ),
-                platform,
-                targets,
+        for freq in freq_range:
+            platform.qubits[qubit].twpa.local_oscillator.frequency = (
+                initial_twpa_freq[qubit] + freq
             )
 
-            classification_result = classification._fit(classification_data)
+            for power in power_range:
+                for qubit in targets:
+                    platform.qubits[qubit].twpa.local_oscillator.power = (
+                        initial_twpa_power[qubit] + power
+                    )
 
-            data.register_qubit(
-                TwpaFrequencyPowerType,
-                (qubit),
-                dict(
-                    freq=np.array(
-                        [platform.qubits[qubit].twpa.local_oscillator.frequency],
-                        dtype=np.float64,
-                    ),
-                    power=np.array(
-                        [platform.qubits[qubit].twpa.local_oscillator.power],
-                        dtype=np.float64,
+                classification_data = classification._acquisition(
+                    classification.SingleShotClassificationParameters.load(
+                        {"nshots": params.nshots}
                     ),
-                    assignment_fidelity=np.array(
-                        [classification_result.assignment_fidelity[qubit]],
+                    platform,
+                    targets,
+                )
+
+                classification_result = classification._fit(classification_data)
+
+                data.register_qubit(
+                    TwpaFrequencyPowerType,
+                    (qubit),
+                    dict(
+                        freq=np.array(
+                            [platform.qubits[qubit].twpa.local_oscillator.frequency],
+                            dtype=np.float64,
+                        ),
+                        power=np.array(
+                            [platform.qubits[qubit].twpa.local_oscillator.power],
+                            dtype=np.float64,
+                        ),
+                        assignment_fidelity=np.array(
+                            [classification_result.assignment_fidelity[qubit]],
+                        ),
+                        angle=np.array([classification_result.rotation_angle[qubit]]),
+                        threshold=np.array([classification_result.threshold[qubit]]),
                     ),
-                    angle=np.array([classification_result.rotation_angle[qubit]]),
-                    threshold=np.array([classification_result.threshold[qubit]]),
-                ),
-            )
+                )
     return data