Skip to content

Commit 16dbd30

Browse files
committed
Change context only behavior to be explicit
1 parent 2f4a9d1 commit 16dbd30

1 file changed

Lines changed: 7 additions & 5 deletions

File tree

haupt/haupt/orchestration/scheduler/resolver.py

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -551,7 +551,7 @@ def resolve_join_runs(j: V1Join):
551551
if edge_k not in params:
552552
params[edge_k] = {}
553553
params[edge_k]["value"] = param_values.get(edge_k)
554-
if current_param.context_only:
554+
if current_param.context_only is not None:
555555
params[edge_k]["contextOnly"] = current_param.context_only
556556
if current_param.connection:
557557
params[edge_k]["connection"] = current_param.connection
@@ -597,10 +597,12 @@ def resolve_params(self):
597597
self._resolve_matrix_params()
598598

599599
def resolve_io(self):
600-
if self.compiled_operation.inputs:
601-
self.run.inputs = {
602-
io.name: io.value for io in self.compiled_operation.inputs
603-
}
600+
inputs = {io.name: io.value for io in self.compiled_operation.contexts or []}
601+
inputs.update(
602+
{io.name: io.value for io in self.compiled_operation.inputs or []}
603+
)
604+
if inputs:
605+
self.run.inputs = inputs
604606
if self.compiled_operation.outputs:
605607
self.run.outputs = {
606608
io.name: io.value for io in self.compiled_operation.outputs

0 commit comments

Comments
 (0)