diff --git a/py_src/taskito/mixins/decorators.py b/py_src/taskito/mixins/decorators.py index 1596246..44d7d82 100644 --- a/py_src/taskito/mixins/decorators.py +++ b/py_src/taskito/mixins/decorators.py @@ -2,9 +2,9 @@ from __future__ import annotations -import asyncio import contextlib import functools +import inspect import os import sys import typing @@ -196,7 +196,7 @@ def decorator(fn: Callable) -> TaskWrapper: functools.update_wrapper(wrapper, fn) # Mark async status for native async dispatch - is_async = asyncio.iscoroutinefunction(fn) + is_async = inspect.iscoroutinefunction(fn) wrapper._taskito_is_async = is_async if is_async: wrapper._taskito_async_fn = fn