[rllib] support local mode (#2795)

This commit is contained in:
Eric Liang
2018-09-02 23:02:19 -07:00
committed by Robert Nishihara
parent 0ac855e061
commit b37a283053
3 changed files with 34 additions and 0 deletions
+20
View File
@@ -0,0 +1,20 @@
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
import unittest
from ray.rllib.agents.ppo import PPOAgent, DEFAULT_CONFIG
import ray
class LocalModeTest(unittest.TestCase):
def testLocal(self):
ray.init(local_mode=True)
cf = DEFAULT_CONFIG.copy()
agent = PPOAgent(cf, "CartPole-v0")
print(agent.train())
if __name__ == "__main__":
unittest.main(verbosity=2)