wehub-resource-sync
38 lines
1.2 KiB
Python
38 lines
1.2 KiB
Python
import logging
|
|
|
|
from ray._private.ray_constants import DEBUG_AUTOSCALING_STATUS_LEGACY
|
|
from ray.experimental.internal_kv import _internal_kv_initialized, _internal_kv_put
|
|
|
|
"""This file provides legacy support for the old info string in order to
|
|
ensure the dashboard's `api/cluster_status` does not break backwards
|
|
compatibilty.
|
|
"""
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def legacy_log_info_string(autoscaler, nodes):
|
|
tmp = "Cluster status: "
|
|
tmp += info_string(autoscaler, nodes)
|
|
tmp += "\n"
|
|
tmp += autoscaler.load_metrics.info_string()
|
|
tmp += "\n"
|
|
tmp += autoscaler.resource_demand_scheduler.debug_string(
|
|
nodes,
|
|
autoscaler.pending_launches.breakdown(),
|
|
autoscaler.load_metrics.get_resource_utilization(),
|
|
)
|
|
if _internal_kv_initialized():
|
|
_internal_kv_put(DEBUG_AUTOSCALING_STATUS_LEGACY, tmp, overwrite=True)
|
|
logger.debug(tmp)
|
|
|
|
|
|
def info_string(autoscaler, nodes):
|
|
suffix = ""
|
|
if autoscaler.updaters:
|
|
suffix += " ({} updating)".format(len(autoscaler.updaters))
|
|
if autoscaler.num_failed_updates:
|
|
suffix += " ({} failed to update)".format(len(autoscaler.num_failed_updates))
|
|
|
|
return "{} nodes{}".format(len(nodes), suffix)
|