diff --git a/influxdb_metrics/tasks.py b/influxdb_metrics/tasks.py index 9d3682a..759abcf 100644 --- a/influxdb_metrics/tasks.py +++ b/influxdb_metrics/tasks.py @@ -4,10 +4,12 @@ try: from celery import shared_task except ImportError: - def shared_task(func): - def wrapper(*args, **kwargs): - return func(*args, **kwargs) - return wrapper + def shared_task(*args, **kwargs): + def decorator(func): + def wrapper(*args, **kwargs): + return func(*args, **kwargs) + return wrapper + return decorator from .utils import write_points as write_points_normal