Skip to content

Commit

Permalink
feat(app): use generic for append_list util
Browse files Browse the repository at this point in the history
  • Loading branch information
psychedelicious committed Feb 27, 2025
1 parent 2a8bff6 commit 4a706d0
Showing 1 changed file with 5 additions and 3 deletions.
8 changes: 5 additions & 3 deletions invokeai/app/invocations/metadata_linked.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down Expand Up @@ -119,8 +119,10 @@
"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.
Expand All @@ -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)
Expand Down

0 comments on commit 4a706d0

Please sign in to comment.