ray/rllib/tests/test_local.py

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import unittest

from ray.rllib.agents.ppo import PPOTrainer, DEFAULT_CONFIG
import ray


class LocalModeTest(unittest.TestCase):
    def testLocal(self):
        ray.init(local_mode=True)
        cf = DEFAULT_CONFIG.copy()
        agent = PPOTrainer(cf, "CartPole-v0")
        print(agent.train())


if __name__ == "__main__":
    unittest.main(verbosity=2)