fix: remove some unused stuff

Author: mrkaye97

hatchet_sdk/v2/hatchet.py

@@ -1,34 +1,12 @@
 import asyncio
 import logging
-from enum import Enum
-from typing import (
-    TYPE_CHECKING,
-    Any,
-    Callable,
-    Generic,
-    Optional,
-    ParamSpec,
-    Type,
-    TypeVar,
-    Union,
-)
-
-from pydantic import BaseModel, ConfigDict
+from typing import TYPE_CHECKING, Any, Callable, Optional, TypeVar
+
 from typing_extensions import deprecated
 
 from hatchet_sdk.clients.rest_client import RestApi
 from hatchet_sdk.context.context import Context
-from hatchet_sdk.contracts.workflows_pb2 import (
-    ConcurrencyLimitStrategy,
-    CreateStepRateLimit,
-    CreateWorkflowJobOpts,
-    CreateWorkflowStepOpts,
-    CreateWorkflowVersionOpts,
-    DesiredWorkerLabels,
-    StickyStrategy,
-    WorkflowConcurrencyOpts,
-    WorkflowKind,
-)
+from hatchet_sdk.contracts.workflows_pb2 import DesiredWorkerLabels
 from hatchet_sdk.features.cron import CronClient
 from hatchet_sdk.features.scheduled import ScheduledClient
 from hatchet_sdk.labels import DesiredWorkerLabel
@@ -48,21 +26,6 @@
 R = TypeVar("R")
 
 
-class HatchetRest:
-    """
-    Main client for interacting with the Hatchet API.
-
-    This class provides access to various client interfaces and utility methods
-    for working with Hatchet via the REST API,
-
-    Attributes:
-        rest (RestApi): Interface for REST API operations.
-    """
-
-    def __init__(self, config: ClientConfig = ClientConfig()):
-        self.rest = RestApi(config.server_url, config.token, config.tenant_id)
-
-
 def transform_desired_worker_label(d: DesiredWorkerLabel) -> DesiredWorkerLabels:
     value = d.value
     return DesiredWorkerLabels(

hatchet_sdk/worker/runner/runner.py

@@ -225,7 +225,9 @@ def inner_callback(task: asyncio.Task[Any]) -> None:
         return inner_callback
 
     ## TODO: Stricter type hinting here
-    def thread_action_func(self, context: Context, step: "Step[T]", action: Action) -> T:
+    def thread_action_func(
+        self, context: Context, step: "Step[T]", action: Action
+    ) -> T:
         if action.step_run_id is not None and action.step_run_id != "":
             self.threads[action.step_run_id] = current_thread()
         elif (