added median sweep and fixed multi device issue

python_app/workflows/reference_workflow.py

@@ -5,16 +5,26 @@ 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_reference_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 reference set."""
-    session = SequentialCaptureSession(config=config, kind="s21_reference", set_name=set_name)
+    session = SequentialCaptureSession(
+        config=config,
+        kind="s21_reference",
+        set_name=set_name,
+        median_sweep_count=median_sweep_count,
+    )
     try:
         session.open()
         while not session.is_complete():