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

Fix flake8

Browse Source
This commit is contained in:
Brandon Rising 2023-08-18 15:47:27 -04:00
parent 4b6d9a73ed
commit 137fbacb92
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,4 @@
from typing import Optional, Union
import networkx as nx import networkx as nx
import copy
from abc import ABC, abstractmethod from abc import ABC, abstractmethod
from itertools import product from itertools import product

9
invokeai/app/services/batch_manager_storage.py
View File

@ -6,7 +6,6 @@ from typing import List, Literal, Optional, Union, cast
from pydantic import BaseModel, Extra, Field, StrictFloat, StrictInt, StrictStr, parse_raw_as, validator from pydantic import BaseModel, Extra, Field, StrictFloat, StrictInt, StrictStr, parse_raw_as, validator
from invokeai.app.services.image_record_storage import OffsetPaginatedResults
from invokeai.app.invocations.primitives import ImageField from invokeai.app.invocations.primitives import ImageField
from invokeai.app.services.graph import Graph from invokeai.app.services.graph import Graph
@ -450,7 +449,7 @@ class SqliteBatchProcessStorage(BatchProcessStorageBase):
"""--sql """--sql
SELECT bp.* SELECT bp.*
FROM batch_process bp FROM batch_process bp
WHERE bp.batch_id IN WHERE bp.batch_id IN
( (
SELECT batch_id SELECT batch_id
FROM batch_session bs FROM batch_session bs
@ -476,7 +475,7 @@ class SqliteBatchProcessStorage(BatchProcessStorageBase):
try: try:
self._lock.acquire() self._lock.acquire()
self._cursor.execute( self._cursor.execute(
f"""--sql """--sql
UPDATE batch_process UPDATE batch_process
SET canceled = 0 SET canceled = 0
WHERE batch_id = ?; WHERE batch_id = ?;
@ -497,7 +496,7 @@ class SqliteBatchProcessStorage(BatchProcessStorageBase):
try: try:
self._lock.acquire() self._lock.acquire()
self._cursor.execute( self._cursor.execute(
f"""--sql """--sql
UPDATE batch_process UPDATE batch_process
SET canceled = 1 SET canceled = 1
WHERE batch_id = ?; WHERE batch_id = ?;
@ -640,7 +639,7 @@ class SqliteBatchProcessStorage(BatchProcessStorageBase):
# Change the state of a batch session # Change the state of a batch session
if changes.state is not None: if changes.state is not None:
self._cursor.execute( self._cursor.execute(
f"""--sql """--sql
UPDATE batch_session UPDATE batch_session
SET state = ? SET state = ?
WHERE batch_id = ? AND session_id = ?; WHERE batch_id = ? AND session_id = ?;