From dca5a2ce261e4a0807581f6d2330c88555446604 Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Sat, 20 Jul 2024 19:57:12 +1000 Subject: [PATCH] fix(app): fix swapped image counts for uncategorized --- invokeai/app/services/board_records/board_records_sqlite.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/invokeai/app/services/board_records/board_records_sqlite.py b/invokeai/app/services/board_records/board_records_sqlite.py index f7848c2195..a9f5605beb 100644 --- a/invokeai/app/services/board_records/board_records_sqlite.py +++ b/invokeai/app/services/board_records/board_records_sqlite.py @@ -295,8 +295,8 @@ class SqliteBoardRecordStorage(BoardRecordStorageBase): """ self._cursor.execute(query) results = self._cursor.fetchall() - image_count = results[0][1] - asset_count = results[1][1] + image_count = dict(results)['images'] + asset_count = dict(results)['assets'] return UncategorizedImageCounts(image_count=image_count, asset_count=asset_count) finally: self._lock.release()