Make sure all the fields in the prediction json files are uploaded
Browse files
utils.py
CHANGED
@@ -21,6 +21,7 @@ def update_model_queue(repo_id, model_name, commit_id, inference_function, statu
|
|
21 |
|
22 |
predictions_object = {
|
23 |
"model_name": model_name,
|
|
|
24 |
"commit_id": commit_id,
|
25 |
"inference_function": inference_function,
|
26 |
"last_updated_timestamp": timestamp,
|
@@ -48,6 +49,7 @@ def upload_predictions(repo_id, predictions, model_name, commit_id, inference_fu
|
|
48 |
predictions_object = {
|
49 |
"model_name": model_name,
|
50 |
"predictions": predictions,
|
|
|
51 |
"inference_function": inference_function,
|
52 |
"last_updated_timestamp": timestamp,
|
53 |
"status": "completed",
|
|
|
21 |
|
22 |
predictions_object = {
|
23 |
"model_name": model_name,
|
24 |
+
"predictions": [],
|
25 |
"commit_id": commit_id,
|
26 |
"inference_function": inference_function,
|
27 |
"last_updated_timestamp": timestamp,
|
|
|
49 |
predictions_object = {
|
50 |
"model_name": model_name,
|
51 |
"predictions": predictions,
|
52 |
+
"commit_id": commit_id,
|
53 |
"inference_function": inference_function,
|
54 |
"last_updated_timestamp": timestamp,
|
55 |
"status": "completed",
|