feat: Implement split batches concatenation

Author: alecandido

src/qibolab/_core/instruments/qblox/cluster.py

@@ -1,6 +1,6 @@
 import time
 from collections import defaultdict
-from functools import cached_property
+from functools import cached_property, reduce
 from itertools import groupby
 from typing import Optional, cast
 
@@ -58,13 +58,26 @@ def batch_shots(
     nfull, remainder = np.divmod(bins * samples, ACQUISITION_MEMORY)
     return (
         [ACQUISITION_MEMORY] * nfull + [remainder]
-        if options.averaging_mode is not AveragingMode.SINGLESHOT
+        if options.averaging_mode is AveragingMode.SINGLESHOT
         else [options.nshots]
     )
 
 
-def concat_shots(results: list[dict[int, Result]]) -> dict[int, Result]:
-    return results[0]
+def concat_shots(
+    results: list[dict[int, Result]], options: ExecutionParameters
+) -> dict[int, Result]:
+    """Concatenate shots batches.
+
+    It collects the measurements' batches resulting from the subdivision generated by
+    the :func:`batch_shots` function.
+    It assumes homogenuous batches, which are only joined over the single outermost
+    dimension, and only for :obj:`AveragingMode.SINGLESHOT` acquisitions.
+    """
+    return (
+        reduce(lambda d, e: {k: np.vstack((v, e[k])) for k, v in d.items()}, results)
+        if options.averaging_mode is AveragingMode.SINGLESHOT
+        else results[0]
+    )
 
 
 class Cluster(Controller):
@@ -218,7 +231,7 @@ def play(
                         options.results_shape(sweepers),
                     )
                 )
-            results |= concat_shots(psres)
+            results |= concat_shots(psres, options)
         return results
 
     def _configure(