From 52bdb1a80c94f8ecf5d3833283b3cd3f078cd245 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Andr=C3=A9=20Heuer?= <mail@andre-heuer.de>
Date: Mon, 21 Aug 2023 21:13:26 +0200
Subject: [PATCH] Added self type

---
 src/documents/tasks.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/documents/tasks.py b/src/documents/tasks.py
index 7d73f852a..0f16b717c 100644
--- 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,
 ):