diff --git a/temporalio/worker/__init__.py b/temporalio/worker/__init__.py index a12bf442b..5a6451dbf 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 ceac29816..17d75db27 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():