From de3f327d7ab5055d0e1154eb924aa564f2590e57 Mon Sep 17 00:00:00 2001 From: Charles Zaloom <38677807+czaloom@users.noreply.github.com> Date: Thu, 1 Aug 2024 20:50:07 -0400 Subject: [PATCH] Finalization Fix (#698) --- api/valor_api/main.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/api/valor_api/main.py b/api/valor_api/main.py index e5484169e..0c64bbf0c 100644 --- 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)