Added self type

diff --git a/src/documents/tasks.py b/src/documents/tasks.py
index 7d73f852af6bc1942943f2475782b7985e10388e..0f16b717c078bbb9c7ed5b2728fa0968ad83fd21 100644 (file)
--- a/src/documents/tasks.py
+++ b/src/documents/tasks.py
@@ -7,6 +7,7 @@ from typing import Type
 
 import tqdm
 from asgiref.sync import async_to_sync
+from celery import Task
 from celery import shared_task
 from channels.layers import get_channel_layer
 from django.conf import settings
@@ -93,7 +94,7 @@ def train_classifier():
 
 @shared_task(bind=True)
 def consume_file(
-    self,
+    self: Task,
     input_doc: ConsumableDocument,
     overrides: Optional[DocumentMetadataOverrides] = None,
 ):