wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-29 14:23:38 +08:00
Code Issues Packages Projects Releases Wiki Activity

[autoscaler] Clean up monitoring loop code (#11677)

Browse Source
This commit is contained in:
Eric Liang
2020-10-30 13:48:43 -07:00
committed by GitHub
parent 6e2a1eac36
commit f9f372c327
2 changed files with 11 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_multi_node_2.py
+2 -2
View File
@@ -68,13 +68,13 @@ def setup_monitor(address):
monitor = Monitor(
address, None, redis_password=ray_constants.REDIS_DEFAULT_PASSWORD)
monitor.update_raylet_map(_append_port=True)
monitor.psubscribe(ray.gcs_utils.XRAY_JOB_PATTERN) # TODO: Remove?
monitor.subscribe(ray.ray_constants.AUTOSCALER_RESOURCE_REQUEST_CHANNEL)
return monitor
def verify_load_metrics(monitor, expected_resource_usage=None, timeout=30):
while True:
monitor.get_all_heartbeat()
monitor.update_load_metrics()
monitor.process_messages()
resource_usage = monitor.load_metrics._get_resource_usage()