From 9cc8b00cb753627fdd77904b697f66f56a93c5ce Mon Sep 17 00:00:00 2001
From: Stavros Efthymiou <35475381+stavros11@users.noreply.github.com>
Date: Wed, 2 Oct 2024 18:04:41 +0400
Subject: [PATCH] feat: updates for Rabi routines

---
 src/qibocal/update.py | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/src/qibocal/update.py b/src/qibocal/update.py
index 215deb43e..b3d0993a0 100644
--- 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(