durabletask 0.0.0.dev70__tar.gz → 0.0.0.dev72__tar.gz

{durabletask-0.0.0.dev70 → durabletask-0.0.0.dev72}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 0.0.0.dev70
+Version: 0.0.0.dev72
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-0.0.0.dev70 → durabletask-0.0.0.dev72}/durabletask/client.py

@@ -224,7 +224,7 @@ class TaskHubGrpcClient:
             method_name: str,
             request: Any,
             *,
-            timeout: Optional[int] = None):
+            timeout: Optional[float] = None):
         method = getattr(self._stub, method_name)
         try:
             if timeout is None:
@@ -406,7 +406,7 @@ class TaskHubGrpcClient:
 
     def wait_for_orchestration_start(self, instance_id: str, *,
                                      fetch_payloads: bool = False,
-                                     timeout: int = 60) -> Optional[OrchestrationState]:
+                                     timeout: float = 60) -> Optional[OrchestrationState]:
         req = pb.GetInstanceRequest(instanceId=instance_id, getInputsAndOutputs=fetch_payloads)
         try:
             self._logger.info(f"Waiting up to {timeout}s for instance '{instance_id}' to start.")
@@ -427,7 +427,7 @@ class TaskHubGrpcClient:
 
     def wait_for_orchestration_completion(self, instance_id: str, *,
                                           fetch_payloads: bool = True,
-                                          timeout: int = 60) -> Optional[OrchestrationState]:
+                                          timeout: float = 60) -> Optional[OrchestrationState]:
         req = pb.GetInstanceRequest(instanceId=instance_id, getInputsAndOutputs=fetch_payloads)
         try:
             self._logger.info(f"Waiting {timeout}s for instance '{instance_id}' to complete.")
@@ -685,7 +685,7 @@ class AsyncTaskHubGrpcClient:
             method_name: str,
             request: Any,
             *,
-            timeout: Optional[int] = None):
+            timeout: Optional[float] = None):
         method = getattr(self._stub, method_name)
         try:
             if timeout is None:
@@ -733,10 +733,9 @@ class AsyncTaskHubGrpcClient:
             await asyncio.sleep(30.0)
             await channel.close()
         finally:
-            try:
-                self._retired_channels.remove(channel)
-            except ValueError:
-                pass
+            async with self._recreate_lock:
+                if channel in self._retired_channels:
+                    self._retired_channels.remove(channel)
 
     async def schedule_new_orchestration(self, orchestrator: Union[task.Orchestrator[TInput, TOutput], str], *,
                                          input: Optional[TInput] = None,
@@ -843,7 +842,7 @@ class AsyncTaskHubGrpcClient:
 
     async def wait_for_orchestration_start(self, instance_id: str, *,
                                            fetch_payloads: bool = False,
-                                           timeout: int = 60) -> Optional[OrchestrationState]:
+                                           timeout: float = 60) -> Optional[OrchestrationState]:
         req = pb.GetInstanceRequest(instanceId=instance_id, getInputsAndOutputs=fetch_payloads)
         try:
             self._logger.info(f"Waiting up to {timeout}s for instance '{instance_id}' to start.")
@@ -863,7 +862,7 @@ class AsyncTaskHubGrpcClient:
 
     async def wait_for_orchestration_completion(self, instance_id: str, *,
                                                 fetch_payloads: bool = True,
-                                                timeout: int = 60) -> Optional[OrchestrationState]:
+                                                timeout: float = 60) -> Optional[OrchestrationState]:
         req = pb.GetInstanceRequest(instanceId=instance_id, getInputsAndOutputs=fetch_payloads)
         try:
             self._logger.info(f"Waiting {timeout}s for instance '{instance_id}' to complete.")

{durabletask-0.0.0.dev70 → durabletask-0.0.0.dev72}/durabletask/internal/grpc_resiliency.py

@@ -13,7 +13,8 @@ def get_full_jitter_delay_seconds(
         attempt: int,
         *,
         base_seconds: float,
-        cap_seconds: float) -> float:
+        cap_seconds: float,
+) -> float:
     capped_attempt = min(attempt, 30)
     upper_bound = min(cap_seconds, base_seconds * (2 ** capped_attempt))
     return random.random() * upper_bound

{durabletask-0.0.0.dev70 → durabletask-0.0.0.dev72}/durabletask/worker.py

@@ -194,7 +194,7 @@ class _InFlightChannelTracker:
         try:
             channel.close()
         except Exception:
-            pass
+            logging.debug("Ignoring channel close failure during worker cleanup.", exc_info=True)
 
 
 class VersioningOptions:
@@ -744,9 +744,10 @@ class TaskHubGrpcWorker:
 
         def wrap_execution(handler, release):
             def wrapped(*args, **kwargs):
-                result = handler(*args, **kwargs)
-                release()
-                return result
+                try:
+                    return handler(*args, **kwargs)
+                finally:
+                    release()
 
             return wrapped
 

{durabletask-0.0.0.dev70 → durabletask-0.0.0.dev72}/durabletask.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: durabletask
-Version: 0.0.0.dev70
+Version: 0.0.0.dev72
 Summary: A Durable Task Client SDK for Python
 License:     MIT License
         

{durabletask-0.0.0.dev70 → durabletask-0.0.0.dev72}/pyproject.toml

@@ -9,7 +9,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "durabletask"
-version = "0.0.0.dev70"
+version = "0.0.0.dev72"
 description = "A Durable Task Client SDK for Python"
 keywords = [
     "durable",