From db323268a2e9db88e6f28efe762ab1a0461910da Mon Sep 17 00:00:00 2001 From: Carson Katri Date: Sun, 20 Nov 2022 18:57:16 -0500 Subject: [PATCH] Use InputGroup default values --- api/tree.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/api/tree.py b/api/tree.py index 73eb4ee..d97287d 100644 --- a/api/tree.py +++ b/api/tree.py @@ -48,9 +48,11 @@ def tree(name): raise Exception(f"Type of tree input '{param.name}' is not a valid 'Type' subclass.") for param in signature.parameters.values(): if issubclass(param.annotation, InputGroup): + instance = param.annotation() prefix = (param.annotation.prefix + "_") if hasattr(param.annotation, "prefix") else "" for group_param, annotation in param.annotation.__annotations__.items(): - inputs[prefix + group_param] = (annotation, inspect.Parameter.empty, param.name, prefix) + default = getattr(instance, group_param, None) + inputs[prefix + group_param] = (annotation, inspect.Parameter.empty if default is None else default, param.name, prefix) else: validate_param(param) inputs[param.name] = (param.annotation, param.default, None, None)