feedback import

Signed-off-by: Filinto Duran <[email protected]>

Author: filintod

durabletask/worker.py

@@ -6,6 +6,8 @@
 import logging
 import os
 import random
+import threading
+import time
 from concurrent.futures import ThreadPoolExecutor
 from datetime import datetime, timedelta
 from threading import Event, Thread
@@ -237,10 +239,8 @@ def __init__(
         self._channel_options = channel_options
         self._stop_timeout = stop_timeout
         # Track in-flight activity executions for graceful draining
-        import threading as _threading
-
         self._active_task_count = 0
-        self._active_task_cv = _threading.Condition()
+        self._active_task_cv = threading.Condition()
 
         # Use provided concurrency options or create default ones
         self._concurrency_options = (
@@ -532,16 +532,13 @@ def wait_for_idle(self, timeout: Optional[float] = None) -> bool:
         """
         end: Optional[float] = None
         if timeout is not None:
-            import time as _t
 
-            end = _t.time() + timeout
+            end = time.time() + timeout
         with self._active_task_cv:
             while self._active_task_count > 0:
                 remaining = None
                 if end is not None:
-                    import time as _t
-
-                    remaining = max(0.0, end - _t.time())
+                    remaining = max(0.0, end - time.time())
                     if remaining == 0.0:
                         return False
                 self._active_task_cv.wait(timeout=remaining)