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

Fix autoscaler format string for memory (#5542)

Browse Source 
* add format string

* fix cast
This commit is contained in:
Eric Liang
2019-08-26 23:25:11 -07:00
committed by GitHub
parent ff73b67154
commit d20696300e
3 changed files with 24 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_autoscaler.py
+12 -3
View File
@@ -174,10 +174,19 @@ class LoadMetricsTest(unittest.TestCase):
lm = LoadMetrics()
lm.update("1.1.1.1", {"CPU": 2}, {"CPU": 0})
lm.update("2.2.2.2", {"CPU": 2, "GPU": 16}, {"CPU": 2, "GPU": 2})
lm.update("3.3.3.3", {
"memory": 20,
"object_store_memory": 40
}, {
"memory": 0,
"object_store_memory": 20
})
debug = lm.info_string()
assert "ResourceUsage=2.0/4.0 CPU, 14.0/16.0 GPU" in debug
assert "NumNodesConnected=2" in debug
assert "NumNodesUsed=1.88" in debug
assert ("ResourceUsage=2.0/4.0 CPU, 14.0/16.0 GPU, "
"1.05 GiB/1.05 GiB memory, "
"1.05 GiB/2.1 GiB object_store_memory") in debug
assert "NumNodesConnected=3" in debug
assert "NumNodesUsed=2.88" in debug
class AutoscalingTest(unittest.TestCase):