Update invokeai/app/services/batch_manager_storage.py

Co-authored-by: psychedelicious <4822129+psychedelicious@users.noreply.github.com>
This commit is contained in:
Brandon 2023-08-15 15:54:17 -04:00 committed by GitHub
parent ed40aee4c5
commit 85105fc070
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -451,7 +451,7 @@ class SqliteBatchProcessStorage(BatchProcessStorageBase):
(batch_id,),
)
result = cast(list[sqlite3.Row], self._cursor.fetchone())
result = cast(Optional[sqlite3.Row], self._cursor.fetchone())
except sqlite3.Error as e:
self._conn.rollback()
raise BatchSessionNotFoundException from e