fix: Update task type to use str instead of enum

pinterest · Aug 13, 2024 · 485d9d0 · 485d9d0
1 parent d91293a
commit 485d9d0
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/querybook/server/tasks/all_tasks.py b/querybook/server/tasks/all_tasks.py
@@ -61,4 +61,4 @@ def configure_workers(sender=None, conf=None, **kwargs):
 @task_failure.connect
 def handle_task_failure(sender, signal, *args, **kwargs):
     task_type = get_schedule_task_type(sender.name)
-    stats_logger.incr(TASK_FAILURES, tags={"task_type": task_type})
+    stats_logger.incr(TASK_FAILURES, tags={"task_type": task_type.value})
diff --git a/querybook/server/tasks/disable_scheduled_docs.py b/querybook/server/tasks/disable_scheduled_docs.py
@@ -65,7 +65,7 @@ def get_scheduled_datadoc_tasks(session=None):
         .filter(
             TaskSchedule.enabled.is_(True),
             TaskSchedule.name.like(DATADOC_SCHEDULE_PREFIX + "%"),
-            TaskSchedule.task_type == ScheduleTaskType.USER,
+            TaskSchedule.task_type == ScheduleTaskType.USER.value,
         )
         .all()
     )