feat: updates for Rabi routines

qiboteam · Oct 11, 2024 · 9cc8b00 · 9cc8b00
1 parent e033aa5
commit 9cc8b00
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/qibocal/update.py b/src/qibocal/update.py
@@ -58,7 +58,9 @@ def drive_amplitude(amp: Union[float, tuple, list], platform: Platform, qubit: Q
     """Update drive frequency value in platform for specific qubit."""
     if isinstance(amp, Iterable):
         amp = amp[0]
-    # platform.natives.single_qubit[qubit].RX.amplitude = float(amp)
+    channel, pulse = platform.natives.single_qubit[qubit].RX[0]
+    new_pulse = replace(pulse, amplitude=amp)
+    platform.natives.single_qubit[qubit].RX[0] = (channel, new_pulse)
 
 
 def drive_duration(
@@ -67,7 +69,9 @@ def drive_duration(
     """Update drive duration value in platform for specific qubit."""
     if isinstance(duration, Iterable):
         duration = duration[0]
-    # platform.natives.single_qubit[qubit].RX.duration = int(duration)
+    channel, pulse = platform.natives.single_qubit[qubit].RX[0]
+    new_pulse = replace(pulse, duration=duration)
+    platform.natives.single_qubit[qubit].RX[0] = (channel, new_pulse)
 
 
 def crosstalk_matrix(