hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

Don't print Redis connection warning in ray.init(). (#8475)

Browse source
This commit is contained in:
Robert Nishihara 2020-05-18 11:19:13 -07:00 • committed by GitHub
parent b6c4f45ae0
commit 2cff471d2c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
python/ray/services.py
View file

@ -527,10 +527,7 @@ def start_ray_process(command,
use_tmux=use_tmux)
def wait_for_redis_to_start(redis_ip_address,
redis_port,
password=None,
num_retries=5):
def wait_for_redis_to_start(redis_ip_address, redis_port, password=None):
"""Wait for a Redis server to be available.
This is accomplished by creating a Redis client and sending a random
@ -540,8 +537,6 @@ def wait_for_redis_to_start(redis_ip_address,
redis_ip_address (str): The IP address of the redis server.
redis_port (int): The port of the redis server.
password (str): The password of the redis server.
num_retries (int): The number of times to try connecting with redis.
The client will sleep for one second between attempts.
Raises:
Exception: An exception is raised if we could not connect with Redis.
@ -549,8 +544,9 @@ def wait_for_redis_to_start(redis_ip_address,
redis_client = redis.StrictRedis(
host=redis_ip_address, port=redis_port, password=password)
# Wait for the Redis server to start.
counter = 0
while counter < num_retries:
num_retries = 12
delay = 0.001
for _ in range(num_retries):
try:
# Run some random command and see if it worked.
logger.debug(
@ -559,12 +555,11 @@ def wait_for_redis_to_start(redis_ip_address,
redis_client.client_list()
except redis.ConnectionError:
# Wait a little bit.
time.sleep(1)
logger.info("Failed to connect to the redis server, retrying.")
counter += 1
time.sleep(delay)
delay *= 2
else:
break
if counter == num_retries:
else:
raise RuntimeError("Unable to connect to Redis. If the Redis instance "
"is on a different machine, check that your "
"firewall is configured properly.")