tests(nodes): fix test instantiation of ImageField

This commit is contained in:
psychedelicious
2023-04-15 16:01:42 +10:00
parent f94d63ec94
commit dfb934a2d4

View File

@@ -49,7 +49,7 @@ class ImageTestInvocation(BaseInvocation):
prompt: str = Field(default = "")
def invoke(self, context: InvocationContext) -> ImageTestInvocationOutput:
return ImageTestInvocationOutput(image=ImageField(image_name=self.id))
return ImageTestInvocationOutput(image=ImageField(image_name=self.id, width=512, height=512, mode="", info={}))
class PromptCollectionTestInvocationOutput(BaseInvocationOutput):
type: Literal['test_prompt_collection_output'] = 'test_prompt_collection_output'