fix: inheritance

Author: mrkaye97

hatchet_sdk/v2/workflows.py

@@ -143,7 +143,7 @@ def __init__(
         self.concurrency__limit_strategy = concurrency__limit_strategy
 
 
-class RegisteredStep(Step[R]):
+class RegisteredStep(Generic[R]):
     def __init__(
         self,
         workflow: "BaseWorkflow",
@@ -153,27 +153,33 @@ def __init__(
         self.step = step
 
     def call(self, ctx: Context) -> R:
-        if self.is_async_function:
-            raise TypeError(f"{self.name} is not a sync function. Use `acall` instead.")
+        if self.step.is_async_function:
+            raise TypeError(
+                f"{self.step.name} is not a sync function. Use `acall` instead."
+            )
 
-        sync_fn = self.fn
+        sync_fn = self.step.fn
         if is_sync_fn(sync_fn):
             return sync_fn(self.workflow, ctx)
 
-        raise TypeError(f"{self.name} is not a sync function. Use `acall` instead.")
+        raise TypeError(
+            f"{self.step.name} is not a sync function. Use `acall` instead."
+        )
 
     async def acall(self, ctx: Context) -> R:
-        if not self.is_async_function:
+        if not self.step.is_async_function:
             raise TypeError(
-                f"{self.name} is not an async function. Use `call` instead."
+                f"{self.step.name} is not an async function. Use `call` instead."
             )
 
-        async_fn = self.fn
+        async_fn = self.step.fn
 
         if is_async_fn(async_fn):
             return await async_fn(self.workflow, ctx)
 
-        raise TypeError(f"{self.name} is not an async function. Use `call` instead.")
+        raise TypeError(
+            f"{self.step.name} is not an async function. Use `call` instead."
+        )
 
 
 class WorkflowDeclaration(Generic[TWorkflowInput]):

hatchet_sdk/worker/runner/runner.py

@@ -241,7 +241,7 @@ async def async_wrapped_action_func(
         sr.set(context.step_run_id)
 
         try:
-            if step.is_async_function:
+            if step.step.is_async_function:
                 return await step.acall(context)
             else:
                 pfunc = functools.partial(