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

Add API for getting total cluster resources. (#1736)

Browse Source 
* Add API for getting total cluster resources.

* Add test.
This commit is contained in:
Robert Nishihara
2018-03-20 15:57:00 -07:00
committed by Philipp Moritz
parent 794f547d0a
commit 2922e1c388
2 changed files with 26 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/runtest.py
+4 -1
View File
@@ -1840,7 +1840,10 @@ class GlobalStateAPI(unittest.TestCase):
with self.assertRaises(Exception):
ray.global_state.log_files()
ray.init()
ray.init(num_cpus=5, num_gpus=3, resources={"CustomResource": 1})
resources = {"CPU": 5, "GPU": 3, "CustomResource": 1}
assert ray.global_state.cluster_resources() == resources
self.assertEqual(ray.global_state.object_table(), dict())