wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 07:50:30 +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()
python/ray/tune/tests/test_cluster.py
+2 -2
View File
@@ -554,7 +554,7 @@ ray.init(address="{address}")
tune.run(
"PG",
name="experiment",
config=dict(env="CartPole-v1"),
config=dict(env="CartPole-v1", framework="tf"),
stop=dict(training_iteration=10),
local_dir="{checkpoint_dir}",
checkpoint_freq=1,
@@ -593,7 +593,7 @@ tune.run(
"experiment": {
"run": "PG",
"checkpoint_freq": 1,
"local_dir": dirpath
"local_dir": dirpath,
}
},
resume=True)
python/ray/tune/tests/test_tune_restore.py
+3
View File
@@ -36,6 +36,7 @@ class TuneRestoreTest(unittest.TestCase):
local_dir=tmpdir,
config={
"env": "CartPole-v0",
"framework": "tf",
},
)
@@ -58,6 +59,7 @@ class TuneRestoreTest(unittest.TestCase):
restore=self.checkpoint_path, # Restore the checkpoint
config={
"env": "CartPole-v0",
"framework": "tf",
},
)
@@ -73,6 +75,7 @@ class TuneRestoreTest(unittest.TestCase):
restore=self.checkpoint_path,
config={
"env": "CartPole-v0",
"framework": "tf",
},
)
self.assertTrue(os.path.isfile(self.checkpoint_path))