@@ -24,7 +24,7 @@ def set_broker(self, broker: "AsyncBroker") -> None:
2424
2525 def startup (
2626 self ,
27- ) -> Union [None , Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" ]:
27+ ) -> Union [Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" , None ]:
2828 """
2929 Startup method to perform various action during startup.
3030
@@ -36,7 +36,7 @@ def startup(
3636
3737 def shutdown (
3838 self ,
39- ) -> Union [None , Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" ]:
39+ ) -> Union [Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" , None ]:
4040 """
4141 Shutdown method to perform various action during shutdown.
4242
@@ -68,7 +68,7 @@ def pre_send(
6868 def post_send (
6969 self ,
7070 message : "TaskiqMessage" ,
71- ) -> Union [None , Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" ]:
71+ ) -> Union [Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" , None ]:
7272 """
7373 This hook is executed right after the task is sent.
7474
@@ -101,7 +101,7 @@ def post_execute(
101101 self ,
102102 message : "TaskiqMessage" ,
103103 result : "TaskiqResult[Any]" ,
104- ) -> Union [None , Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" ]:
104+ ) -> Union [Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" , None ]:
105105 """
106106 This hook executes after task is complete.
107107
@@ -116,7 +116,7 @@ def post_save(
116116 self ,
117117 message : "TaskiqMessage" ,
118118 result : "TaskiqResult[Any]" ,
119- ) -> Union [None , Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" ]:
119+ ) -> Union [Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" , None ]:
120120 """
121121 Post save hook.
122122
@@ -132,7 +132,7 @@ def on_error(
132132 message : "TaskiqMessage" ,
133133 result : "TaskiqResult[Any]" ,
134134 exception : BaseException ,
135- ) -> Union [None , Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" ]:
135+ ) -> Union [Coroutine [Any , Any , None ], "CoroutineType[Any, Any, None]" , None ]:
136136 """
137137 This function is called when exception is found.
138138
0 commit comments