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

[RLlib] Fix broken tune tests in master due to framework=auto errors. (#8672)

Browse Source
This commit is contained in:
Sven Mika
2020-05-29 11:55:47 +02:00
committed by GitHub
parent c64b694560
commit d483ed28ba
6 changed files with 18 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/tests/test_memory_scheduling.py
+4
View File
@@ -77,6 +77,7 @@ class TestMemoryScheduling(unittest.TestCase):
config={
"env": "CartPole-v0",
"memory": 100 * 1024 * 1024, # too little
"framework": "tf",
},
raise_on_failed_trial=False)
self.assertEqual(result.trials[0].status, "ERROR")
@@ -99,6 +100,7 @@ class TestMemoryScheduling(unittest.TestCase):
"env": "CartPole-v0",
# too large
"object_store_memory": 10000 * 1024 * 1024,
"framework": "tf",
}))
finally:
ray.shutdown()
@@ -113,6 +115,7 @@ class TestMemoryScheduling(unittest.TestCase):
"env": "CartPole-v0",
"num_workers": 1,
"memory_per_worker": 100 * 1024 * 1024, # too little
"framework": "tf",
},
raise_on_failed_trial=False)
self.assertEqual(result.trials[0].status, "ERROR")
@@ -134,6 +137,7 @@ class TestMemoryScheduling(unittest.TestCase):
"num_workers": 1,
# too large
"object_store_memory_per_worker": 10000 * 1024 * 1024,
"framework": "tf",
}))
finally:
ray.shutdown()