fix test definition and improve code for flipping with rx90 gate

qiboteam · Dec 6, 2024 · b90f887 · b90f887
1 parent af82fec
commit b90f887
Show file tree

Hide file tree

Showing 3 changed files with 78 additions and 85 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/qibocal/protocols/flipping.py b/src/qibocal/protocols/flipping.py
@@ -36,34 +36,27 @@ def flipping_sequence(
 
     if rx90:
         sequence |= natives.RX90()
-
-        for _ in range(flips):
-            qd_channel, qd_pulse = natives.RX90()[0]
-
-            qd_detuned = update.replace(
-                qd_pulse, delta_amplitude=qd_pulse.amplitude + delta_amplitude
-            )
-
-            sequence.append((qd_channel, qd_detuned))
-            sequence.append((qd_channel, qd_detuned))
-            sequence.append((qd_channel, qd_detuned))
-            sequence.append((qd_channel, qd_detuned))
-
-        sequence |= natives.MZ()
-
     else:
         sequence |= natives.R(theta=np.pi / 2)
 
-        for _ in range(flips):
+    for _ in range(flips):
+
+        if rx90:
+            qd_channel, qd_pulse = natives.RX90()[0]
+        else:
             qd_channel, qd_pulse = natives.RX()[0]
 
-            qd_detuned = update.replace(
-                qd_pulse, amplitude=qd_pulse.amplitude + delta_amplitude
-            )
+        qd_detuned = update.replace(
+            qd_pulse, delta_amplitude=qd_pulse.amplitude + delta_amplitude
+        )
+        sequence.append((qd_channel, qd_detuned))
+        sequence.append((qd_channel, qd_detuned))
+
+        if rx90:
             sequence.append((qd_channel, qd_detuned))
             sequence.append((qd_channel, qd_detuned))
 
-        sequence |= natives.MZ()
+    sequence |= natives.MZ()
 
     return sequence