fix(api): update enum usage to work for python 3.11

dunkeroni · Oct 12, 2024 · 80360a8 · 80360a8
1 parent acfeb4a
commit 80360a8
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/invokeai/app/services/board_records/board_records_sqlite.py b/invokeai/app/services/board_records/board_records_sqlite.py
@@ -168,7 +168,9 @@ def get_many(
             # Determine archived filter condition
             archived_filter = "" if include_archived else "WHERE archived = 0"
 
-            final_query = base_query.format(archived_filter=archived_filter, order_by=order_by, direction=direction)
+            final_query = base_query.format(
+                archived_filter=archived_filter, order_by=order_by.value, direction=direction.value
+            )
 
             # Execute query to fetch boards
             self._cursor.execute(final_query, (limit, offset))
@@ -208,7 +210,7 @@ def get_all(
         try:
             self._lock.acquire()
 
-            if order_by == "board_name":
+            if order_by == BoardRecordOrderBy.Name:
                 base_query = """
                     SELECT *
                     FROM boards
@@ -225,7 +227,9 @@ def get_all(
 
             archived_filter = "" if include_archived else "WHERE archived = 0"
 
-            final_query = base_query.format(archived_filter=archived_filter, order_by=order_by, direction=direction)
+            final_query = base_query.format(
+                archived_filter=archived_filter, order_by=order_by.value, direction=direction.value
+            )
 
             self._cursor.execute(final_query)