added median sweep and fixed multi device issue

python_app/workflows/calibration_workflow.py

@@ -5,13 +5,18 @@ from __future__ import annotations
 from python_app.models.dataset_model import SweepCollection
 from python_app.models.run_config_model import RunConfigModel
 from python_app.storage.npz_store import NpzStore
-from python_app.workflows.sequential_capture_workflow import SequentialCaptureSession
+from python_app.workflows.sequential_capture_workflow import (
+    DEFAULT_CALIBRATION_MEDIAN_SWEEP_COUNT,
+    SequentialCaptureSession,
+)
 
 
 def capture_calibration_set(
     config: RunConfigModel,
     set_name: str,
     store: NpzStore,
+    *,
+    median_sweep_count: int = DEFAULT_CALIBRATION_MEDIAN_SWEEP_COUNT,
 ) -> tuple[str, SweepCollection]:
     """Capture all switch combinations and persist them as calibration set."""
     if config.is_multi_device:
@@ -21,7 +26,12 @@ def capture_calibration_set(
             "and captured explicitly."
         )
 
-    session = SequentialCaptureSession(config=config, kind="s21_calibration", set_name=set_name)
+    session = SequentialCaptureSession(
+        config=config,
+        kind="s21_calibration",
+        set_name=set_name,
+        median_sweep_count=median_sweep_count,
+    )
     try:
         session.open()
         while not session.is_complete():