fix task decorator

Author: snopoke

taskbadger/decorators.py

@@ -14,7 +14,7 @@ def track(
     name: str = None,
     monitor_id: str = None,
     max_runtime: int = None,
-    **kwargs,
+    **task_kwargs,
 ):
     """
     Decorator to track a function as a task.
@@ -50,7 +50,7 @@ def _inner(*args, **kwargs):
                 status=StatusEnum.PROCESSING,
                 max_runtime=max_runtime,
                 monitor_id=monitor_id,
-                **kwargs,
+                **task_kwargs,
             )
             try:
                 result = func(*args, **kwargs)

tests/test_decorators.py

@@ -20,15 +20,19 @@ def test(arg):
 
 @mock.patch("taskbadger.decorators.create_task_safe")
 def test_track_decorator_args(create):
-    @track(name="new name", monitor_id="test", max_runtime=1)
+    @track(name="new name", monitor_id="test", max_runtime=1, tags={"name": "value"})
     def test(arg):
         return arg
 
     assert test("test") == "test"
     assert create.call_count == 1
     assert create.call_args.args[0] == "new name"
-    assert create.call_args.kwargs["monitor_id"] == "test"
-    assert create.call_args.kwargs["max_runtime"] == 1
+    assert create.call_args.kwargs == {
+        "status": "processing",
+        "monitor_id": "test",
+        "max_runtime": 1,
+        "tags": {"name": "value"},
+    }
 
 
 @mock.patch("taskbadger.decorators.create_task_safe")