Merge pull request #42 from scaleapi/sasha/adjust_params

Sasha/adjust params

Author: sasha-scale

nucleus/__init__.py

@@ -92,6 +92,7 @@
 )
 from .constants import (
     NUCLEUS_ENDPOINT,
+    DEFAULT_NETWORK_TIMEOUT_SEC,
     ERRORS_KEY,
     ERROR_ITEMS,
     ERROR_PAYLOAD,
@@ -658,7 +659,7 @@ def predict(
             Union[BoxPrediction, PolygonPrediction, SegmentationPrediction]
         ],
         update: bool,
-        batch_size: int = 100,
+        batch_size: int = 5000,
     ):
         """
         Uploads model outputs as predictions for a model_run. Returns info about the upload.
@@ -1043,6 +1044,7 @@ def _make_grequest(
                 session=sess,
                 files=payload,
                 auth=(self.api_key, ""),
+                timeout=DEFAULT_NETWORK_TIMEOUT_SEC,
             )
         else:
             post = requests_command(
@@ -1051,6 +1053,7 @@ def _make_grequest(
                 json=payload,
                 headers={"Content-Type": "application/json"},
                 auth=(self.api_key, ""),
+                timeout=DEFAULT_NETWORK_TIMEOUT_SEC,
             )
         return post
 
@@ -1074,6 +1077,7 @@ def _make_request_raw(
             json=payload,
             headers={"Content-Type": "application/json"},
             auth=(self.api_key, ""),
+            timeout=DEFAULT_NETWORK_TIMEOUT_SEC,
         )
         logger.info("API request has response code %s", response.status_code)
 

nucleus/constants.py

@@ -1,4 +1,5 @@
 NUCLEUS_ENDPOINT = "https://api.scale.com/v1/nucleus"
+DEFAULT_NETWORK_TIMEOUT_SEC = 60
 ITEMS_KEY = "items"
 ITEM_KEY = "item"
 REFERENCE_ID_KEY = "reference_id"