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

chore(backend): remove unnecessary batch validation function

Browse Source 
The `Batch` model is fully validated by pydantic on instantiation; we do not need any validation logic for it.
This commit is contained in:
psychedelicious 2023-08-17 11:59:47 +10:00
parent c91621b46c
commit 8cf9bd47b2
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
invokeai/app/services/batch_manager.py
View File

@ -112,18 +112,11 @@ class BatchManager(BatchManagerBase):
ges = self.__invoker.services.graph_execution_manager.get(created_session.session_id)
self.__invoker.invoke(ges, invoke_all=True)
def _valid_batch_config(self, batch_process: BatchProcess) -> bool:
# TODO: Check that the node_ids in the batch are unique
# TODO: Validate data types are correct for each batch data
return True
def create_batch_process(self, batch: Batch, graph: Graph) -> BatchProcessResponse:
batch_process = BatchProcess(
batch=batch,
graph=graph,
)
if not self._valid_batch_config(batch_process):
return None
batch_process = self.__batch_process_storage.save(batch_process)
sessions = self._create_sessions(batch_process)
return BatchProcessResponse(