fix: cache tasks args (#871)

codecov · Nov 8, 2024 · 68da17c · 68da17c
1 parent dbd2417
commit 68da17c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tasks/cache_test_rollups.py b/tasks/cache_test_rollups.py
@@ -90,7 +90,7 @@ def get_query(with_end: bool) -> str:
 
 
 class CacheTestRollupsTask(BaseCodecovTask, name=cache_test_rollups_task_name):
-    def run_impl(self, *, repoid, branch, **kwargs):
+    def run_impl(self, _db_session, repoid: int, branch: str, **kwargs):
         redis_conn = get_redis_connection()
         try:
             with redis_conn.lock(

diff --git a/tasks/cache_test_rollups_redis.py b/tasks/cache_test_rollups_redis.py
@@ -13,7 +13,9 @@
 class CacheTestRollupsRedisTask(
     BaseCodecovTask, name=cache_test_rollups_redis_task_name
 ):
-    def run_impl(self, *, repoid: int, branch: str, **kwargs) -> dict[str, bool]:
+    def run_impl(
+        self, _db_session, repoid: int, branch: str, **kwargs
+    ) -> dict[str, bool]:
         redis_conn = get_redis_connection()
         try:
             with redis_conn.lock(