Ray cluster CRD and example CR + multi-ray-cluster operator (#12098)

This commit is contained in:
Gekho457
2020-12-14 10:26:01 -06:00
committed by GitHub
parent 35f7d84dbe
commit 11ce1dc743
21 changed files with 5163 additions and 385 deletions
+9 -3
View File
@@ -85,7 +85,11 @@ class Monitor:
This is used to receive notifications about failed components.
"""
def __init__(self, redis_address, autoscaling_config, redis_password=None):
def __init__(self,
redis_address,
autoscaling_config,
redis_password=None,
prefix_cluster_info=False):
# Initialize the Redis clients.
ray.state.state._initialize_global_state(
redis_address, redis_password=redis_password)
@@ -107,8 +111,10 @@ class Monitor:
head_node_ip = redis_address.split(":")[0]
self.load_metrics = LoadMetrics(local_ip=head_node_ip)
if autoscaling_config:
self.autoscaler = StandardAutoscaler(autoscaling_config,
self.load_metrics)
self.autoscaler = StandardAutoscaler(
autoscaling_config,
self.load_metrics,
prefix_cluster_info=prefix_cluster_info)
self.autoscaling_config = autoscaling_config
else:
self.autoscaler = None