fixed typehint

nens · Mar 12, 2024 · de848a4 · de848a4
1 parent 1be6d95
commit de848a4
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/api/bro_upload/delivery.py b/api/bro_upload/delivery.py
@@ -1,6 +1,7 @@
 import time
 import traceback
 
+from .. import models
 from . import config, utils
 
 
@@ -33,7 +34,7 @@ class BRODelivery:
 
     def __init__(
         self,
-        upload_task_instance: str,
+        upload_task_instance: models.UploadTask,
         bro_username: str,
         bro_password: str,
     ) -> None: