fix: status

andreiionutdamian · Apr 22, 2024 · 5560957 · 5560957
1 parent a34c955
commit 5560957
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/app/src/serving/serving_app.py b/app/src/serving/serving_app.py
@@ -158,6 +158,8 @@ def appmon_callback(self):
           result=json.dumps(result), 
           status=STATUS_FINISHED
         )
+
+        #TODO improve
         self.postgres_update_data(
           table_name="jobs",
           identifier={"uuid":jobid},
@@ -356,6 +358,7 @@ def _predict_job (self, model_type: str, input, params:dict = None):
           if result is None:
             try:
               workername = workername if tndx == 0 else None
+              status =  status if tndx == 0  else STATUS_CREATED
               self.postgres_insert_data(
                   "tasks",
                   uuid = taskid,