AMR-KELEG commited on
Commit
73178f8
·
1 Parent(s): eae4d81

Make sure all the fields in the prediction json files are uploaded

Browse files
Files changed (1) hide show
  1. utils.py +2 -0
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",