Found more places where WorkerType could be specified

minvws · Dec 19, 2024 · 1fcd593 · 1fcd593
1 parent 1742ff0
commit 1fcd593
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/boefjes/boefjes/app.py b/boefjes/boefjes/app.py
@@ -49,7 +49,7 @@ def __init__(
 
         self.exited = False
 
-    def run(self, queue_type: str) -> None:
+    def run(self, queue_type: WorkerManager.WorkerType) -> None:
         logger.info("Created worker pool for queue '%s'", queue_type)
 
         self.workers = [
@@ -80,7 +80,7 @@ def run(self, queue_type: str) -> None:
 
                 raise
 
-    def _fill_queue(self, task_queue: Queue, queue_type: str) -> None:
+    def _fill_queue(self, task_queue: Queue, queue_type: WorkerManager.WorkerType) -> None:
         if task_queue.qsize() > self.settings.pool_size:
             time.sleep(self.settings.worker_heartbeat)
             return