make decorator return wrapped function

miguelteixeiraa · Dec 14, 2023 · 7c655f0 · 7c655f0
1 parent 715e091
commit 7c655f0
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/crinkle/flow.py b/crinkle/flow.py
@@ -49,7 +49,7 @@ def decorator(
             function_processor: Union[
                 TypeFunctionProcessor, TypeFunctionProcessorAsync
             ],
-        ) -> None:
+        ) -> Union[TypeFunctionProcessor, TypeFunctionProcessorAsync]:
             if asyncio.iscoroutinefunction(function_processor):
                 self.add_processor(
                     processor=FunctionProcessorAsync(func=function_processor, name=name)
@@ -59,6 +59,7 @@ def decorator(
             self.add_processor(
                 processor=FunctionProcessor(func=function_processor, name=name)
             )
+            return function_processor
 
         return decorator