[cli] ray memory: added redis_password (#9492)

This commit is contained in:
jerrylee.io
2020-07-18 00:08:41 -07:00
committed by GitHub
parent 7e3ded7439
commit 4c18463fb7
+8 -2
View File
@@ -1120,12 +1120,18 @@ def statistics(address):
required=False,
type=str,
help="Override the address to connect to.")
def memory(address):
@click.option(
"--redis_password",
required=False,
type=str,
default=ray_constants.REDIS_DEFAULT_PASSWORD,
help="Connect to ray with redis_password.")
def memory(address, redis_password):
"""Print object references held in a Ray cluster."""
if not address:
address = services.find_redis_address_or_die()
logger.info("Connecting to Ray instance at {}.".format(address))
ray.init(address=address)
ray.init(address=address, redis_password=redis_password)
print(ray.internal.internal_api.memory_summary())