From 8b1ec2685f0216c6ed4dddae8791a3555a05defd Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Wed, 16 Aug 2023 11:38:47 +1000 Subject: [PATCH] chore: black --- invokeai/app/services/image_record_storage.py | 8 ++++---- invokeai/app/services/models/image_record.py | 2 +- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/invokeai/app/services/image_record_storage.py b/invokeai/app/services/image_record_storage.py index 53d792a577..aa71431b7f 100644 --- a/invokeai/app/services/image_record_storage.py +++ b/invokeai/app/services/image_record_storage.py @@ -67,7 +67,7 @@ IMAGE_DTO_COLS = ", ".join( "created_at", "updated_at", "deleted_at", - "starred" + "starred", ], ) ) @@ -140,7 +140,7 @@ class ImageRecordStorageBase(ABC): node_id: Optional[str], metadata: Optional[dict], is_intermediate: bool = False, - starred: bool = False + starred: bool = False, ) -> datetime: """Saves an image record.""" pass @@ -204,7 +204,7 @@ class SqliteImageRecordStorage(ImageRecordStorageBase): self._cursor.execute("PRAGMA table_info(images)") columns = [column[1] for column in self._cursor.fetchall()] - + if "starred" not in columns: self._cursor.execute( """--sql @@ -529,7 +529,7 @@ class SqliteImageRecordStorage(ImageRecordStorageBase): node_id: Optional[str], metadata: Optional[dict], is_intermediate: bool = False, - starred: bool = False + starred: bool = False, ) -> datetime: try: metadata_json = None if metadata is None else json.dumps(metadata) diff --git a/invokeai/app/services/models/image_record.py b/invokeai/app/services/models/image_record.py index e386c33d1f..a480fd0efd 100644 --- a/invokeai/app/services/models/image_record.py +++ b/invokeai/app/services/models/image_record.py @@ -132,5 +132,5 @@ def deserialize_image_record(image_dict: dict) -> ImageRecord: updated_at=updated_at, deleted_at=deleted_at, is_intermediate=is_intermediate, - starred=starred + starred=starred, )