Skip to content

Commit db32326

Browse files
committed
Use InputGroup default values
1 parent 7001a17 commit db32326

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

api/tree.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -48,9 +48,11 @@ def validate_param(param):
4848
raise Exception(f"Type of tree input '{param.name}' is not a valid 'Type' subclass.")
4949
for param in signature.parameters.values():
5050
if issubclass(param.annotation, InputGroup):
51+
instance = param.annotation()
5152
prefix = (param.annotation.prefix + "_") if hasattr(param.annotation, "prefix") else ""
5253
for group_param, annotation in param.annotation.__annotations__.items():
53-
inputs[prefix + group_param] = (annotation, inspect.Parameter.empty, param.name, prefix)
54+
default = getattr(instance, group_param, None)
55+
inputs[prefix + group_param] = (annotation, inspect.Parameter.empty if default is None else default, param.name, prefix)
5456
else:
5557
validate_param(param)
5658
inputs[param.name] = (param.annotation, param.default, None, None)

0 commit comments

Comments
 (0)