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

make InvocationQueueItem serializable

Browse Source
This commit is contained in:
Eugene Brodsky 2023-04-13 00:23:15 -04:00
parent 50eb02f68b
commit 6f6de402ad
1 changed files with 5 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,31 +1,17 @@
# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)
from abc import ABC, abstractmethod
from queue import Queue
import time
from abc import ABC, abstractmethod
from dataclasses import dataclass
from queue import Queue
# TODO: make this serializable
@dataclass
class InvocationQueueItem:
# session_id: str
graph_execution_state_id: str
invocation_id: str
invoke_all: bool
timestamp: float
def __init__(
self,
# session_id: str,
graph_execution_state_id: str,
invocation_id: str,
invoke_all: bool = False,
):
# self.session_id = session_id
self.graph_execution_state_id = graph_execution_state_id
self.invocation_id = invocation_id
self.invoke_all = invoke_all
self.timestamp = time.time()
timestamp: float = time.time()
class InvocationQueueABC(ABC):
"""Abstract base class for all invocation queues"""