Coverage for mindsdb / utilities / ml_task_queue / base.py: 75%

1from mindsdb.utilities.ml_task_queue.utils import wait_redis_ping 

2 

3 

4class BaseRedisQueue: 

5 def wait_redis_ping(self, timeout: int = 30) -> None: 

6 """ wait when redis.ping return True 

7 

8 Args: 

9 timeout (int): seconds to wait for success ping 

10 

11 Raises: 

12 RedisConnectionError: if `ping` did not return `True` within `timeout` seconds 

13 """ 

14 return wait_redis_ping(self.db, timeout)