1- from typing import List , Dict
1+ from typing import List , Dict , Set
22
33import redis
44import valkey
@@ -87,7 +87,7 @@ def get_connection(queue_settings, use_strict_redis=False):
8787
8888
8989def get_queue (
90- name = "default" , default_timeout = None , is_async = None , autocommit = None , connection = None , ** kwargs
90+ name = "default" , default_timeout = None , is_async = None , autocommit = None , connection = None , ** kwargs
9191) -> DjangoQueue :
9292 """Returns an DjangoQueue using parameters defined in `SCHEDULER_QUEUES`"""
9393 from .settings import QUEUES
@@ -107,18 +107,18 @@ def get_queue(
107107 )
108108
109109
110- def get_all_workers ():
110+ def get_all_workers () -> Set [ DjangoWorker ] :
111111 from .settings import QUEUES
112112
113- workers = set ()
113+ workers_set : Set [ DjangoWorker ] = set ()
114114 for queue_name in QUEUES :
115115 connection = get_connection (QUEUES [queue_name ])
116116 try :
117- curr_workers = set (DjangoWorker .all (connection = connection ))
118- workers .update (curr_workers )
117+ curr_workers : Set [ DjangoWorker ] = set (DjangoWorker .all (connection = connection ))
118+ workers_set .update (curr_workers )
119119 except (redis .ConnectionError , valkey .ConnectionError ) as e :
120120 logger .error (f"Could not connect for queue { queue_name } : { e } " )
121- return workers
121+ return workers_set
122122
123123
124124def _queues_share_connection_params (q1_params : Dict , q2_params : Dict ):
0 commit comments