diff --git a/library-standard/src/mas/library/standard/plugins/design_patterns/linear.py b/library-standard/src/mas/library/standard/plugins/design_patterns/linear.py index 6f43087f..2aaa8381 100644 --- a/library-standard/src/mas/library/standard/plugins/design_patterns/linear.py +++ b/library-standard/src/mas/library/standard/plugins/design_patterns/linear.py @@ -18,7 +18,7 @@ class DeterministicLinearPlugin(_DeterministicBase): plugin_id = "deterministic_linear@v1" mode = "linear" - def evaluate_next(self, q: QProduct, run: RunLedger, *, config: KernelConfig) -> list[EgressSymbol]: + def evaluate_next(self, q: QProduct, run: RunLedger, config: KernelConfig) -> list[EgressSymbol]: st = self._state() if not st.participants: return self._no_participants(q) diff --git a/runtime/src/mas/runtime/kernel/ingress_step.py b/runtime/src/mas/runtime/kernel/ingress_step.py index 52398451..266e9c56 100644 --- a/runtime/src/mas/runtime/kernel/ingress_step.py +++ b/runtime/src/mas/runtime/kernel/ingress_step.py @@ -94,7 +94,7 @@ def commit_engine_io_return( q.dp = DpState.EVALUATING apply_control_valid(q) coord_after_ingress(q) - return evaluate(q, run, config) + return evaluate(q, run, config=config) def apply_engine_io_return( @@ -203,6 +203,6 @@ def apply_engine_io_return( ) q.inflight_kind = "NONE" q.dp = DpState.EVALUATING - return evaluate(q, run, config) + return evaluate(q, run, config=config) return commit_engine_io_return(q, run, event, config=config, evaluate=evaluate) diff --git a/runtime/tests/test_parallel_tools.py b/runtime/tests/test_parallel_tools.py index 7bbeab41..80d9933c 100644 --- a/runtime/tests/test_parallel_tools.py +++ b/runtime/tests/test_parallel_tools.py @@ -102,7 +102,7 @@ def test_schedule_parallel_tools_egress_siblings_share_one_parent() -> None: text="ok", ) with runtime_binding(None, op): - apply_engine_io_return(q, run, event, config=config, evaluate=lambda *_: []) + apply_engine_io_return(q, run, event, config=config, evaluate=lambda *_, **__: []) ends = [ e