chore(item-storage): improve types

Provide type args to the generics.
This commit is contained in:
psychedelicious
2024-02-03 18:35:46 +11:00
committed by Kent Keirsey
parent 88c08bbfc7
commit 9f274c79dc
2 changed files with 6 additions and 6 deletions

View File

@ -10,7 +10,7 @@ from invokeai.app.services.item_storage.item_storage_common import ItemNotFoundE
T = TypeVar("T", bound=BaseModel)
class ItemStorageMemory(ItemStorageABC, Generic[T]):
class ItemStorageMemory(ItemStorageABC[T], Generic[T]):
"""
Provides a simple in-memory storage for items, with a maximum number of items to store.
The storage uses the LRU strategy to evict items from storage when the max has been reached.