fix: make RemoteExecutor context manager non-blocking on pending futures

Author: Namrata Madan

src/sagemaker/remote_function/client.py

@@ -731,7 +731,7 @@ def map(self, func, *iterables):
         futures = map(self.submit, itertools.repeat(func), *iterables)
         return [future.result() for future in futures]
 
-    def shutdown(self):
+    def shutdown(self, wait=True):
         """Prevent more function executions to be submitted to this executor."""
         with self._state_condition:
             self._shutdown = True
@@ -742,15 +742,15 @@ def shutdown(self):
             self._state_condition.notify_all()
 
         if self._workers is not None:
-            self._workers.shutdown(wait=True)
+            self._workers.shutdown(wait)
 
     def __enter__(self):
         """Create an executor instance and return it"""
         return self
 
     def __exit__(self, exc_type, exc_val, exc_tb):
         """Make sure the executor instance is shutdown."""
-        self.shutdown()
+        self.shutdown(wait=False)
         return False
 
     @staticmethod