Worker code cleanup (#692)

temporalio · Nov 22, 2024 · 97a2b7a · 97a2b7a
1 parent 042e088
commit 97a2b7a
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/temporalio/worker/__init__.py b/temporalio/worker/__init__.py
@@ -91,4 +91,13 @@
     "ResourceBasedSlotSupplier",
     "ResourceBasedTunerConfig",
     "ResourceBasedSlotConfig",
+    "ActivitySlotInfo",
+    "CustomSlotSupplier",
+    "LocalActivitySlotInfo",
+    "SlotInfo",
+    "SlotMarkUsedContext",
+    "SlotPermit",
+    "SlotReleaseContext",
+    "SlotReserveContext",
+    "WorkflowSlotInfo",
 ]
diff --git a/tests/worker/test_worker.py b/tests/worker/test_worker.py
@@ -2,7 +2,6 @@
 
 import asyncio
 import concurrent.futures
-import threading
 import uuid
 from datetime import timedelta
 from typing import Any, Awaitable, Callable, Optional
@@ -30,7 +29,7 @@
     WorkflowSlotInfo,
 )
 from temporalio.workflow import VersioningIntent
-from tests.helpers import assert_eq_eventually, new_worker, worker_versioning_enabled
+from tests.helpers import new_worker, worker_versioning_enabled
 
 
 def test_load_default_worker_binary_id():