Finalization Fix (#698)

Striveworks · Aug 2, 2024 · de3f327 · de3f327
1 parent b091f55
commit de3f327
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/api/valor_api/main.py b/api/valor_api/main.py
@@ -764,9 +764,7 @@ def finalize_dataset(
 
     """
     try:
-        crud.finalize(
-            db=db, dataset_name=dataset_name, task_handler=background_tasks
-        )
+        crud.finalize(db=db, dataset_name=dataset_name, task_handler=None)
     except Exception as e:
         raise exceptions.create_http_error(e)
 
@@ -1280,7 +1278,7 @@ def finalize_inferences(
             db=db,
             model_name=model_name,
             dataset_name=dataset_name,
-            task_handler=background_tasks,
+            task_handler=None,
         )
     except Exception as e:
         raise exceptions.create_http_error(e)