trying to make formatter happy

scaleapi · Dec 18, 2024 · 9c918fa · 9c918fa
1 parent 53a38e9
commit 9c918fa
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/tests/test_prediction.py b/tests/test_prediction.py
@@ -561,9 +561,10 @@ def test_default_category_pred_upload_update(model_run):
 
     # Copy so we don't modify the original.
     prediction_update_params = dict(TEST_DEFAULT_CATEGORY_PREDICTIONS[1])
-    prediction_update_params["reference_id"] = (
-        TEST_DEFAULT_CATEGORY_PREDICTIONS[0]["reference_id"]
-    )
+    prediction_update_params[
+        "reference_id"
+    ] = TEST_DEFAULT_CATEGORY_PREDICTIONS[0]["reference_id"]
+
 
     prediction_update = CategoryPrediction.from_json(prediction_update_params)
     response = model_run.predict(annotations=[prediction_update], update=True)
@@ -588,9 +589,9 @@ def test_default_category_pred_upload_ignore(model_run):
 
     # Copy so we don't modify the original.
     prediction_update_params = dict(TEST_DEFAULT_CATEGORY_PREDICTIONS[1])
-    prediction_update_params["reference_id"] = (
-        TEST_DEFAULT_CATEGORY_PREDICTIONS[0]["reference_id"]
-    )
+    prediction_update_params[
+        "reference_id"
+    ] = TEST_DEFAULT_CATEGORY_PREDICTIONS[0]["reference_id"]
 
     prediction_update = CategoryPrediction.from_json(prediction_update_params)
     # Default behavior is ignore.