wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-07-19 11:27:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Improve cluster.wait_for_nodes() API. (#3712)

Browse Source 
* Separate out functionality for querying client table and improve cluster.wait_for_nodes() API.

* Linting

* Add back logging statements.

* info -> debug
This commit is contained in:
Robert Nishihara
2019-01-07 21:26:58 -08:00
committed by Philipp Moritz
parent 33319502b6
commit 5e76d52868
8 changed files with 130 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/multi_node_test_2.py
+3 -2
View File
@@ -5,6 +5,7 @@ from __future__ import print_function
import json
import logging
import pytest
import time
import ray
import ray.services as services
@@ -82,10 +83,10 @@ def test_internal_config(start_connected_longer_cluster):
cluster.wait_for_nodes()
cluster.remove_node(worker)
cluster.wait_for_nodes(retries=10)
time.sleep(1)
assert ray.global_state.cluster_resources()["CPU"] == 2
cluster.wait_for_nodes(retries=20)
time.sleep(2)
assert ray.global_state.cluster_resources()["CPU"] == 1