def main(args):
# Astar Module
if args.model == "astar":
model = Astar()
# Reinforcement Learning module
elif args.model == "rl":
model = Rl()
# Dummy test
elif args.model == "dummy":
model = Dummy()
# Create an environment
env = gym.make('voilier-v2').unwrapped
state = env.reset()
# TODO Get range from argument
for step in range(0, 200, 1):
action = model.step(state)
state, _, _, _ = env.step(action)
env.render()
