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",
|