modify atmos pred

key-value 구조로 return값을 변경하면서 .tolist()로 인해 오류가 나던 부분을 수정했습니다.
State-of-The-MLOps · Oct 9, 2021 · 22d57cb · 22d57cb
1 parent 7ef78f0
commit 22d57cb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/app/api/router/predict.py b/app/api/router/predict.py
@@ -83,11 +83,11 @@ def sync_pred_ts(time_series):
             f"Predict Args info: {time_series.flatten().tolist()}\n\tmodel_name: {my_model.model_name}\n\tPrediction Result: {result.tolist()[0]}"
         )
 
-        return {"result": result, "error": None}
+        return {"result": result.tolist(), "error": None}
 
     try:
         result = await run_in_threadpool(sync_pred_ts, time_series)
-        return result.tolist()
+        return result
 
     except Exception as e:
         L.error(e)