fix: Fix allxy related tests

Tests were failing after running them with qibolab 0.1.4. I've fixed them after applying the same fix already present in allXY protocol.
qiboteam · Dec 7, 2023 · d1112d7 · d1112d7
1 parent b44c94c
commit d1112d7
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/src/qibocal/protocols/characterization/allxy/allxy_drag_pulse_tuning.py b/src/qibocal/protocols/characterization/allxy/allxy_drag_pulse_tuning.py
@@ -94,7 +94,9 @@ def _acquisition(
                 # store the results
                 gate = "-".join(gates)
                 data.register_qubit(
-                    allxy.AllXYType, (qubit, beta_param), dict(prob=z_proj, gate=gate)
+                    allxy.AllXYType,
+                    (qubit, beta_param),
+                    dict(prob=np.array([z_proj]), gate=np.array([gate])),
                 )
     return data
 

diff --git a/src/qibocal/protocols/characterization/allxy/drag_pulse_tuning.py b/src/qibocal/protocols/characterization/allxy/drag_pulse_tuning.py
@@ -143,8 +143,8 @@ def _acquisition(
                 DragPulseTuningType,
                 (qubit),
                 dict(
-                    signal=r1.magnitude - r2.magnitude,
-                    beta=beta_param,
+                    signal=np.array([r1.magnitude - r2.magnitude]),
+                    beta=np.array([beta_param]),
                 ),
             )