mirror of
https://github.com/wassname/ray.git
synced 2026-07-09 01:03:34 +08:00
Sven
60d4d5e1aa
* Remove all __future__ imports from RLlib. * Remove (object) again from tf_run_builder.py::TFRunBuilder. * Fix 2xLINT warnings. * Fix broken appo_policy import (must be appo_tf_policy) * Remove future imports from all other ray files (not just RLlib). * Remove future imports from all other ray files (not just RLlib). * Remove future import blocks that contain `unicode_literals` as well. Revert appo_tf_policy.py to appo_policy.py (belongs to another PR). * Add two empty lines before Schedule class. * Put back __future__ imports into determine_tests_to_run.py. Fails otherwise on a py2/print related error.
78 lines
2.2 KiB
Python
78 lines
2.2 KiB
Python
import unittest
|
|
|
|
import ray
|
|
from ray.rllib import _register_all
|
|
from ray.tune import register_trainable
|
|
from ray.tune.experiment import Experiment, convert_to_experiment_list
|
|
from ray.tune.error import TuneError
|
|
|
|
|
|
class ExperimentTest(unittest.TestCase):
|
|
def tearDown(self):
|
|
ray.shutdown()
|
|
_register_all() # re-register the evicted objects
|
|
|
|
def setUp(self):
|
|
def train(config, reporter):
|
|
for i in range(100):
|
|
reporter(timesteps_total=i)
|
|
|
|
register_trainable("f1", train)
|
|
|
|
def testConvertExperimentFromExperiment(self):
|
|
exp1 = Experiment(**{
|
|
"name": "foo",
|
|
"run": "f1",
|
|
"config": {
|
|
"script_min_iter_time_s": 0
|
|
}
|
|
})
|
|
result = convert_to_experiment_list(exp1)
|
|
self.assertEqual(len(result), 1)
|
|
self.assertEqual(type(result), list)
|
|
|
|
def testConvertExperimentNone(self):
|
|
result = convert_to_experiment_list(None)
|
|
self.assertEqual(len(result), 0)
|
|
self.assertEqual(type(result), list)
|
|
|
|
def testConvertExperimentList(self):
|
|
exp1 = Experiment(**{
|
|
"name": "foo",
|
|
"run": "f1",
|
|
"config": {
|
|
"script_min_iter_time_s": 0
|
|
}
|
|
})
|
|
result = convert_to_experiment_list([exp1, exp1])
|
|
self.assertEqual(len(result), 2)
|
|
self.assertEqual(type(result), list)
|
|
|
|
def testConvertExperimentJSON(self):
|
|
experiment = {
|
|
"name": {
|
|
"run": "f1",
|
|
"config": {
|
|
"script_min_iter_time_s": 0
|
|
}
|
|
},
|
|
"named": {
|
|
"run": "f1",
|
|
"config": {
|
|
"script_min_iter_time_s": 0
|
|
}
|
|
}
|
|
}
|
|
result = convert_to_experiment_list(experiment)
|
|
self.assertEqual(len(result), 2)
|
|
self.assertEqual(type(result), list)
|
|
|
|
def testConvertExperimentIncorrect(self):
|
|
self.assertRaises(TuneError, lambda: convert_to_experiment_list("hi"))
|
|
|
|
|
|
if __name__ == "__main__":
|
|
import pytest
|
|
import sys
|
|
sys.exit(pytest.main(["-v", __file__]))
|