Finish burning down MyPy backlog

conftest.py

@@ -3,7 +3,7 @@
 import time
 from io import BytesIO
 from threading import Thread
-from typing import AsyncGenerator, Callable, cast
+from typing import AsyncGenerator, Callable, Generator, cast
 
 import psutil
 import pytest
@@ -23,7 +23,9 @@ def hatchet() -> Hatchet:
 
 
 @pytest.fixture()
-def worker(request: pytest.FixtureRequest):
+def worker(
+    request: pytest.FixtureRequest,
+) -> Generator[subprocess.Popen[bytes], None, None]:
     example = cast(str, request.param)
 
     command = ["poetry", "run", example]

examples/bulk_fanout/bulk_trigger.py

@@ -7,7 +7,7 @@
 from dotenv import load_dotenv
 
 from hatchet_sdk import new_client
-from hatchet_sdk.clients.admin import TriggerWorkflowOptions
+from hatchet_sdk.clients.admin import TriggerWorkflowOptions, WorkflowRunDict
 from hatchet_sdk.clients.rest.models.workflow_run import WorkflowRun
 from hatchet_sdk.clients.run_event_listener import StepRunEventType
 
@@ -16,22 +16,20 @@ async def main() -> None:
     load_dotenv()
     hatchet = new_client()
 
-    workflowRuns: list[dict[str, Any]] = []
-
     # we are going to run the BulkParent workflow 20 which will trigger the Child workflows n times for each n in range(20)
-    for i in range(20):
-        workflowRuns.append(
-            {
-                "workflow_name": "BulkParent",
-                "input": {"n": i},
-                "options": {
-                    "additional_metadata": {
-                        "bulk-trigger": i,
-                        "hello-{i}": "earth-{i}",
-                    },
-                },
-            }
+    workflowRuns = [
+        WorkflowRunDict(
+            workflow_name="BulkParent",
+            input={"n": i},
+            options=TriggerWorkflowOptions(
+                additional_metadata={
+                    "bulk-trigger": str(i),
+                    f"hello-{i}": f"earth-{i}",
+                }
+            ),
         )
+        for i in range(20)
+    ]
 
     workflowRunRefs = hatchet.admin.run_workflows(
         workflowRuns,

hatchet_sdk/__init__.py

@@ -137,8 +137,9 @@
 from .clients.run_event_listener import StepRunEventType, WorkflowRunEventType
 from .context.context import Context
 from .context.worker_context import WorkerContext
-from .hatchet import ClientConfig, Hatchet, concurrency, on_failure_step, step, workflow
-from .worker import Worker, WorkerStartOptions, WorkerStatus
+from .hatchet import Hatchet, concurrency, on_failure_step, step, workflow
+from .loader import ClientConfig
+from .worker.worker import Worker, WorkerStartOptions, WorkerStatus
 from .workflow import ConcurrencyExpression
 
 __all__ = [

hatchet_sdk/client.py

@@ -30,7 +30,7 @@ def from_environment(
         defaults: ClientConfig = ClientConfig(),
         debug: bool = False,
         *opts_functions: Callable[[ClientConfig], None],
-    ):
+    ) -> "Client":
         try:
             loop = asyncio.get_running_loop()
         except RuntimeError:
@@ -48,7 +48,7 @@ def from_config(
         cls,
         config: ClientConfig = ClientConfig(),
         debug: bool = False,
-    ):
+    ) -> "Client":
         try:
             loop = asyncio.get_running_loop()
         except RuntimeError:
@@ -89,7 +89,7 @@ def __init__(
         rest_client: RestApi,
         config: ClientConfig,
         debug: bool = False,
-    ):
+    ) -> None:
         try:
             loop = asyncio.get_running_loop()
         except RuntimeError:
@@ -107,8 +107,8 @@ def __init__(
         self.debug = debug
 
 
-def with_host_port(host: str, port: int):
-    def with_host_port_impl(config: ClientConfig):
+def with_host_port(host: str, port: int) -> Callable[[ClientConfig], None]:
+    def with_host_port_impl(config: ClientConfig) -> None:
         config.host = host
         config.port = port