File tree Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Expand file tree Collapse file tree 1 file changed +7
-6
lines changed Original file line number Diff line number Diff line change @@ -561,9 +561,10 @@ def test_default_category_pred_upload_update(model_run):
561
561
562
562
# Copy so we don't modify the original.
563
563
prediction_update_params = dict (TEST_DEFAULT_CATEGORY_PREDICTIONS [1 ])
564
- prediction_update_params ["reference_id" ] = (
565
- TEST_DEFAULT_CATEGORY_PREDICTIONS [0 ]["reference_id" ]
566
- )
564
+ prediction_update_params [
565
+ "reference_id"
566
+ ] = TEST_DEFAULT_CATEGORY_PREDICTIONS [0 ]["reference_id" ]
567
+
567
568
568
569
prediction_update = CategoryPrediction .from_json (prediction_update_params )
569
570
response = model_run .predict (annotations = [prediction_update ], update = True )
@@ -588,9 +589,9 @@ def test_default_category_pred_upload_ignore(model_run):
588
589
589
590
# Copy so we don't modify the original.
590
591
prediction_update_params = dict (TEST_DEFAULT_CATEGORY_PREDICTIONS [1 ])
591
- prediction_update_params ["reference_id" ] = (
592
- TEST_DEFAULT_CATEGORY_PREDICTIONS [ 0 ][ "reference_id" ]
593
- )
592
+ prediction_update_params [
593
+ "reference_id"
594
+ ] = TEST_DEFAULT_CATEGORY_PREDICTIONS [ 0 ][ "reference_id" ]
594
595
595
596
prediction_update = CategoryPrediction .from_json (prediction_update_params )
596
597
# Default behavior is ignore.
You can’t perform that action at this time.
0 commit comments