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)