Skip to content

Commit cf8272b

Browse files
committed
🐛 Add imports to guardrail node initialization
1 parent 07a91cd commit cf8272b

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

CPAC/registration/guardrails.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -111,6 +111,10 @@ def registration_guardrail_node(name=None):
111111
return Node(Function(input_names=['registered',
112112
'reference'],
113113
output_names=['registered'],
114+
imports=['from CPAC.qc import qc_masks, '
115+
'REGISTRATION_GUARDRAIL_THRESHOLDS',
116+
'from CPAC.registration.guardrails '
117+
'import BadRegistrationError'],
114118
function=registration_guardrail), name=name)
115119

116120

0 commit comments

Comments
 (0)