git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-04-20 13:45:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat(app): use generic for append_list util

Browse Source
This commit is contained in:
psychedelicious 2025-02-27 09:36:34 +10:00
parent 2a8bff601f
commit 4a706d00bb
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/invocations/metadata_linked.py
View File

@ -3,7 +3,7 @@
# Thanks to @skunworkxdark for the original implementation!
import copy
from typing import Any, Dict, Literal, Optional, Union
from typing import Any, Dict, Literal, Optional, TypeVar, Union
from pydantic import model_validator
@ -119,8 +119,10 @@ CORE_LABELS_VAE = Literal[
"vae",
]
T = TypeVar("T")
def append_list(item_cls: type[Any], new_item: Any, items: Union[Any, list[Any], None] = None) -> list[Any]:
def append_list(item_cls: type[T], new_item: T, items: Union[T, list[T], None] = None) -> list[T]:
"""Combines any number of items or lists into a single list,
ensuring consistency in type.
@ -142,7 +144,7 @@ def append_list(item_cls: type[Any], new_item: Any, items: Union[Any, list[Any],
if items is None:
return [new_item]
result: list[item_cls] = []
result: list[T] = []
if isinstance(items, item_cls):
result.append(items)