Unpack kwargs when awaiting runs in BaselineAlgorithm and RemoteAlgor…

…ithm (#134)
SneaksAndData · Jan 27, 2025 · 01b578c · 01b578c
1 parent e57a82d
commit 01b578c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/esd_services_api_client/nexus/algorithms/_baseline_algorithm.py b/esd_services_api_client/nexus/algorithms/_baseline_algorithm.py
@@ -87,6 +87,7 @@ async def _measured_run(**run_args):
 
         return await partial(
             _measured_run,
+            **kwargs,
             **self._inputs,
             metric_tags=self._metric_tags,
             metrics_provider=self._metrics_provider,

diff --git a/esd_services_api_client/nexus/algorithms/_remote_algorithm.py b/esd_services_api_client/nexus/algorithms/_remote_algorithm.py
@@ -113,6 +113,7 @@ async def _measured_run(**run_args) -> AlgorithmResult:
 
         return await partial(
             _measured_run,
+            **kwargs,
             **results,
             metric_tags=self._metric_tags,
             metrics_provider=self._metrics_provider,