Skip to content
Snippets Groups Projects
Commit ec337400 authored by David Hammer's avatar David Hammer
Browse files

Merge branch 'feat/kernel-decision-in-output' into 'master'

Decision mask in arbter kernel output schema

See merge request !124
parents 4869c69a b8829419
No related branches found
No related tags found
1 merge request!124Decision mask in arbter kernel output schema
...@@ -397,6 +397,7 @@ class AdvancedFrameSelectionArbiter(BaseFrameSelectionArbiter): ...@@ -397,6 +397,7 @@ class AdvancedFrameSelectionArbiter(BaseFrameSelectionArbiter):
except Exception as ex: except Exception as ex:
warn(f"Kernel for {prefix} failed: {ex}") warn(f"Kernel for {prefix} failed: {ex}")
res[selection_name] = np.ones(num_frames, dtype=bool) res[selection_name] = np.ones(num_frames, dtype=bool)
out_hash[f"{selection_name}.mask"] = list(map(bool, res[selection_name]))
return res return res
def _configure_decision(self, decision_string): def _configure_decision(self, decision_string):
...@@ -475,6 +476,17 @@ class AdvancedFrameSelectionArbiter(BaseFrameSelectionArbiter): ...@@ -475,6 +476,17 @@ class AdvancedFrameSelectionArbiter(BaseFrameSelectionArbiter):
kernel_class.extend_device_schema( kernel_class.extend_device_schema(
schema_update, kernel_prefix schema_update, kernel_prefix
) )
(
NODE_ELEMENT(output_schema_update)
.key(selection_name)
.commit(),
VECTOR_BOOL_ELEMENT(output_schema_update)
.key(f"{selection_name}.mask")
.assignmentOptional()
.defaultValue([])
.commit(),
)
kernel_class.extend_output_schema(output_schema_update, selection_name) kernel_class.extend_output_schema(output_schema_update, selection_name)
new_selection_steps[selection_name] = (kernel_class, preselection) new_selection_steps[selection_name] = (kernel_class, preselection)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment