Dump report after each protocol

src/qibocal/auto/execute.py

@@ -141,3 +141,4 @@ def run(self):
             if self.platform is not None:
                 if self.update and task.update:
                     self.platform.update(completed.results.update)
+            yield

src/qibocal/cli/_base.py

@@ -59,7 +59,6 @@ def command(runcard, folder, force, update):
     builder.run()
     if update:
         builder.dump_platform_runcard()
-    builder.dump_report()
 
 
 @click.command(context_settings=CONTEXT_SETTINGS)

src/qibocal/cli/builders.py

@@ -95,13 +95,13 @@ def _prepare_output(self, runcard):
 
     def run(self):
         """Execute protocols in runcard."""
-
         if self.platform is not None:
             self.platform.connect()
             self.platform.setup()
             self.platform.start()
 
-        self.executor.run()
+        for _ in self.executor.run():
+            self.dump_report()
 
         if self.platform is not None:
             self.platform.stop()

src/qibocal/protocols/characterization/resonator_spectroscopy.py

@@ -1,4 +1,4 @@
-from dataclasses import dataclass, field
+from dataclasses import dataclass, field, fields
 from typing import Optional, Union
 
 import numpy as np
@@ -9,7 +9,14 @@
 from qibolab.qubits import QubitId
 from qibolab.sweeper import Parameter, Sweeper, SweeperType
 
-from qibocal.auto.operation import Data, Parameters, Qubits, Results, Routine
+from qibocal.auto.operation import (
+    Data,
+    Parameters,
+    ParameterValue,
+    Qubits,
+    Results,
+    Routine,
+)
 
 from .utils import PowerLevel, lorentzian_fit, spectroscopy_plot
 
@@ -61,6 +68,22 @@ class ResonatorSpectroscopyResults(Results):
     )
     """Readout attenuation [dB] for each qubit."""
 
+    @property
+    def update(self):
+        """Method overwritten from Results to not update
+        amplitude when running resonator spectroscopy at
+        high power."""
+        up: dict[str, ParameterValue] = {}
+        if self.bare_frequency == {}:
+            fields_to_updated = [fld for fld in fields(self) if fld.name != "amplitude"]
+        else:
+            fields_to_updated = fields(self)
+        for fld in fields_to_updated:
+            if "update" in fld.metadata:
+                up[fld.metadata["update"]] = getattr(self, fld.name)
+
+        return up
+
 
 ResSpecType = np.dtype(
     [("freq", np.float64), ("msr", np.float64), ("phase", np.float64)]

src/qibocal/protocols/characterization/utils.py

@@ -90,7 +90,6 @@ def spectroscopy_plot(data, fit: Results, qubit):
     qubit_data = data[qubit]
 
     fitting_report = ""
-
     frequencies = qubit_data.freq * HZ_TO_GHZ
     fig.add_trace(
         go.Scatter(
@@ -145,14 +144,9 @@ def spectroscopy_plot(data, fit: Results, qubit):
     else:
         label = "qubit frequency"
         freq = fit.frequency
-
     fitting_report += f"{qubit} | {label}: {freq[qubit]*GHZ_TO_HZ:,.0f} Hz<br>"
-
     if fit.amplitude[qubit] is not None:
         fitting_report += f"{qubit} | amplitude: {fit.amplitude[qubit]} <br>"
-        if data.power_level is PowerLevel.high:
-            # TODO: find better solution for not updating amplitude in high power
-            fit.amplitude.pop(qubit)
 
     if data.__class__.__name__ == "ResonatorSpectroscopyAttenuationData":
         if fit.attenuation[qubit] is not None and fit.attenuation[qubit] != 0:

tests/test_auto.py

@@ -36,6 +36,7 @@ def test_execution(card: pathlib.Path):
     """
     testcard = TestCard(**yaml.safe_load(card.read_text(encoding="utf-8")))
     executor = Executor.load(testcard.runcard, output=pathlib.Path(tempfile.mkdtemp()))
-    executor.run()
+    for _ in executor.run():
+        pass
 
     assert testcard.validation.result == [step[0] for step in executor.history]

tests/test_task_options.py

@@ -34,7 +34,7 @@
                 "freq_width": 10_000_000,
                 "freq_step": 100_000,
                 "amplitude": 0.4,
-                "power_level": "low",
+                "power_level": "high",
             },
         },
     ],
@@ -99,7 +99,7 @@ def test_update_argument(global_update, local_update):
         platform.qubits,
         global_update,
     )
-    executor.run()
+    next(executor.run())
     if local_update and global_update:
         assert old_readout_frequency != platform.qubits[0].readout_frequency
     else: