wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-28 23:30:16 +08:00
Code Issues Actions 13 Packages Projects Releases Wiki Activity

Fix example in unit test

Browse Source 
This wouldn't have caused any issues in the unit test, but it would have
bugged the UI if someone copy+pasted it into their own node pack.
This commit is contained in:
Jacob Segal 2024-03-23 16:49:45 -07:00
parent 6b6a93cc5d
commit 03394ace8c
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/inference/testing_nodes/testing-pack/tools.py
View File

@ -1,4 +1,9 @@
def MakeSmartType(t):
if isinstance(t, str):
return SmartType(t)
return t
class SmartType(str): class SmartType(str):
def __ne__(self, other): def __ne__(self, other):
if self == "*" or other == "*": if self == "*" or other == "*":
@ -18,12 +23,12 @@ def VariantSupport():
continue continue
for key, value in types[category].items(): for key, value in types[category].items():
if isinstance(value, tuple): if isinstance(value, tuple):
types[category][key] = (SmartType(value[0]),) + value[1:] types[category][key] = (MakeSmartType(value[0]),) + value[1:]
return types return types
setattr(cls, "INPUT_TYPES", new_input_types) setattr(cls, "INPUT_TYPES", new_input_types)
if hasattr(cls, "RETURN_TYPES"): if hasattr(cls, "RETURN_TYPES"):
old_return_types = cls.RETURN_TYPES old_return_types = cls.RETURN_TYPES
setattr(cls, "RETURN_TYPES", tuple(SmartType(x) for x in old_return_types)) setattr(cls, "RETURN_TYPES", tuple(MakeSmartType(x) for x in old_return_types))
if hasattr(cls, "VALIDATE_INPUTS"): if hasattr(cls, "VALIDATE_INPUTS"):
# Reflection is used to determine what the function signature is, so we can't just change the function signature # Reflection is used to determine what the function signature is, so we can't just change the function signature
raise NotImplementedError("VariantSupport does not support VALIDATE_INPUTS yet") raise NotImplementedError("VariantSupport does not support VALIDATE_INPUTS yet")
@ -39,7 +44,7 @@ def VariantSupport():
expected_type = inputs["optional"][key][0] expected_type = inputs["optional"][key][0]
else: else:
expected_type = None expected_type = None
if expected_type is not None and SmartType(value) != expected_type: if expected_type is not None and MakeSmartType(value) != expected_type:
return f"Invalid type of {key}: {value} (expected {expected_type})" return f"Invalid type of {key}: {value} (expected {expected_type})"
return True return True
setattr(cls, "VALIDATE_INPUTS", validate_inputs) setattr(cls, "VALIDATE_INPUTS", validate_inputs)