impose schema structure on clickify_parameters input

caracal-pipeline · Apr 10, 2024 · 866e018 · 866e018
1 parent 6f26e28
commit 866e018
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/scabha/schema_utils.py b/scabha/schema_utils.py
@@ -148,6 +148,9 @@ def clickify_parameters(schemas: Union[str, Dict[str, Any]]):
     if type(schemas) is str:
         schemas = OmegaConf.merge(OmegaConf.structured(Schema),
                                 OmegaConf.load(schemas))
+    else:
+        schemas = OmegaConf.merge(OmegaConf.structured(Schema),
+                                  dict(inputs=schemas.inputs, outputs=schemas.outputs))
 
     decorator_chain = None
     inputs = Cargo.flatten_schemas(OrderedDict(), schemas.inputs, "inputs")