Merge pull request #23 from scaleapi/grequests-no-session

removed session from both local + remote upload

Author: Nastia39

nucleus/__init__.py

@@ -119,7 +119,6 @@ def get_model_run(self, model_run_id: str) -> ModelRun:
         """
         return ModelRun(model_run_id, self)
 
-
     def delete_model_run(self, model_run_id: str):
         """
         Fetches a model_run for given id
@@ -220,7 +219,6 @@ def preprocess_payload(item):
             }
             return payload
 
-        session = requests.session()
         items = payload[ITEMS_KEY]
         responses: List[Any] = []
         for i in range(0, len(items), batch_size):
@@ -231,7 +229,6 @@ def preprocess_payload(item):
                 self._make_grequest(
                     payload,
                     f"dataset/{dataset_id}/append",
-                    session=session,
                     local=True,
                 )
                 for payload in payloads
@@ -273,7 +270,6 @@ def exception_handler(request, exception):
             logger.error(request, exception)
             return default_error(request.json())
 
-        session = requests.session()
         items = payload[ITEMS_KEY]
         payloads = [
             {ITEMS_KEY: items[i : i + batch_size]}
@@ -284,7 +280,6 @@ def exception_handler(request, exception):
             self._make_grequest(
                 payload,
                 f"dataset/{dataset_id}/append",
-                session=session,
                 local=False,
             )
             for payload in payloads
@@ -613,7 +608,7 @@ def _make_grequest(
         self,
         payload: dict,
         route: str,
-        session,
+        session=None,
         requests_command: Callable = grequests.post,
         local=True,
     ):