def test_visualizer(self): try: ray.init(num_cpus=1) # , redis_address="localhost:6379") except Exception: pass # current path current_path = os.path.realpath(__file__).rsplit('/', 1)[0] # run the experiment and check it doesn't crash arg_str = '{}/../data/rllib_data/ 1 --num_rollouts 1 ' \ '--no_render'.format(current_path).split() parser = vs_rllib.create_parser() pass_args = parser.parse_args(arg_str) visualizer_rllib(pass_args)
def test_visualizer_multi(self): """Test for multi-agent visualization""" try: ray.init(num_cpus=1) except Exception: pass # current path current_path = os.path.realpath(__file__).rsplit('/', 1)[0] # run the experiment and check it doesn't crash arg_str = '{}/../data/rllib_data/multi_agent 1 --num-rollouts 1 ' \ '--render_mode no_render ' \ '--horizon 10'.format(current_path).split() parser = vs_rllib.create_parser() pass_args = parser.parse_args(arg_str) visualizer_rllib(pass_args)