Add constants for successful expected http status codes

Author: rkaplan

conftest.py

@@ -15,6 +15,7 @@
 import pytest
 
 import nucleus
+from nucleus.constants import SUCCESS_STATUS_CODES
 
 
 assert 'NUCLEUS_PYTEST_API_KEY' in os.environ, \
@@ -47,7 +48,7 @@ def _make_request_patch(
         payload: dict, route: str, requests_command=requests.post
     ) -> dict:
         response = client._make_request_raw(payload, route, requests_command)
-        assert response.status_code in [200, 201], \
+        assert response.status_code in SUCCESS_STATUS_CODES, \
             f"HTTP response had status code: {response.status_code}. " \
             f"Full JSON: {response.json()}"
         return response.json()

nucleus/__init__.py

@@ -96,6 +96,7 @@
     ANNOTATIONS_PROCESSED_KEY,
     PREDICTIONS_PROCESSED_KEY,
     STATUS_CODE_KEY,
+    SUCCESS_STATUS_CODES,
     DATASET_NAME_KEY,
     DATASET_MODEL_RUNS_KEY,
     DATASET_SLICES_KEY,
@@ -917,7 +918,7 @@ def _make_request(
         """
         response = self._make_request_raw(payload, route, requests_command)
 
-        if response.status_code not in [200, 201]:
+        if response.status_code not in SUCCESS_STATUS_CODES:
             logger.warning(response)
 
         return (

nucleus/constants.py

@@ -16,6 +16,7 @@
 ANNOTATIONS_PROCESSED_KEY = "annotations_processed"
 PREDICTIONS_PROCESSED_KEY = "predictions_processed"
 STATUS_CODE_KEY = "status_code"
+SUCCESS_STATUS_CODES = [200, 201]
 ERRORS_KEY = "errors"
 MODEL_RUN_ID_KEY = "model_run_id"
 MODEL_ID_KEY = "model_id"