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

nodes: convert InvocationQueueItem to Pydantic class

Browse Source
This commit is contained in:
Eugene 2023-04-13 03:44:44 -04:00 committed by Eugene Brodsky
parent 6f6de402ad
commit 7fc5fbd4ce
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
invokeai/app/services/invocation_queue.py
View File

@ -2,16 +2,16 @@
import time import time
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from dataclasses import dataclass from pydantic import BaseModel, Field
from queue import Queue from queue import Queue
@dataclass class InvocationQueueItem(BaseModel):
class InvocationQueueItem:
graph_execution_state_id: str graph_execution_state_id: str
invocation_id: str invocation_id: str
invoke_all: bool invoke_all: bool
timestamp: float = time.time() timestamp: float = Field(default_factory=time.time)
class InvocationQueueABC(ABC): class InvocationQueueABC(ABC):
"""Abstract base class for all invocation queues""" """Abstract base class for all invocation queues"""