Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(tests): add versions to test nodes

Browse Source 
Fixes a warning about missing version.
This commit is contained in:
psychedelicious 2023-12-09 14:32:26 +11:00
parent 0ac33f36ef
commit 8f6e2c0c85
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tests/nodes/test_nodes.py
View File

@ -18,7 +18,7 @@ class ListPassThroughInvocationOutput(BaseInvocationOutput):
collection: list[ImageField] = OutputField(default=[])
@invocation("test_list")
@invocation("test_list", version="1.0.0")
class ListPassThroughInvocation(BaseInvocation):
collection: list[ImageField] = InputField(default=[])
@ -31,7 +31,7 @@ class PromptTestInvocationOutput(BaseInvocationOutput):
prompt: str = OutputField(default="")
@invocation("test_prompt")
@invocation("test_prompt", version="1.0.0")
class PromptTestInvocation(BaseInvocation):
prompt: str = InputField(default="")
@ -39,7 +39,7 @@ class PromptTestInvocation(BaseInvocation):
return PromptTestInvocationOutput(prompt=self.prompt)
@invocation("test_error")
@invocation("test_error", version="1.0.0")
class ErrorInvocation(BaseInvocation):
def invoke(self, context: InvocationContext) -> PromptTestInvocationOutput:
raise Exception("This invocation is supposed to fail")
@ -50,7 +50,7 @@ class ImageTestInvocationOutput(BaseInvocationOutput):
image: ImageField = OutputField()
@invocation("test_text_to_image")
@invocation("test_text_to_image", version="1.0.0")
class TextToImageTestInvocation(BaseInvocation):
prompt: str = InputField(default="")
prompt2: str = InputField(default="")
@ -59,7 +59,7 @@ class TextToImageTestInvocation(BaseInvocation):
return ImageTestInvocationOutput(image=ImageField(image_name=self.id))
@invocation("test_image_to_image")
@invocation("test_image_to_image", version="1.0.0")
class ImageToImageTestInvocation(BaseInvocation):
prompt: str = InputField(default="")
image: Union[ImageField, None] = InputField(default=None)
@ -73,7 +73,7 @@ class PromptCollectionTestInvocationOutput(BaseInvocationOutput):
collection: list[str] = OutputField(default=[])
@invocation("test_prompt_collection")
@invocation("test_prompt_collection", version="1.0.0")
class PromptCollectionTestInvocation(BaseInvocation):
collection: list[str] = InputField()
@ -86,7 +86,7 @@ class AnyTypeTestInvocationOutput(BaseInvocationOutput):
value: Any = OutputField()
@invocation("test_any")
@invocation("test_any", version="1.0.0")
class AnyTypeTestInvocation(BaseInvocation):
value: Any = InputField(default=None)
@ -94,7 +94,7 @@ class AnyTypeTestInvocation(BaseInvocation):
return AnyTypeTestInvocationOutput(value=self.value)
@invocation("test_polymorphic")
@invocation("test_polymorphic", version="1.0.0")
class PolymorphicStringTestInvocation(BaseInvocation):
value: Union[str, list[str]] = InputField(default="")