From 97a2b7a2911879269784dc4549a87372b47b100a Mon Sep 17 00:00:00 2001
From: Dan Davison <dan.davison@temporal.io>
Date: Thu, 21 Nov 2024 20:21:03 -0500
Subject: [PATCH] Worker code cleanup (#692)

---
 temporalio/worker/__init__.py | 9 +++++++++
 tests/worker/test_worker.py   | 3 +--
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/temporalio/worker/__init__.py b/temporalio/worker/__init__.py
index a12bf442..5a6451db 100644
--- 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
index ceac2981..17d75db2 100644
--- 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():