# coding: utf-8
import gym

from agent import AIAgent

# TODO: usd Enduro in gym
def main():
    agent = AIAgent()
    env = gym.Env()


if __name__ == '__main__':
    main()

# coding: utf-8
import gym

from agent import AIAgent

#TODO: usd Enduro in gym
def main():
    agent = AIAgent()
    env = gym.Env()


if __name__ == '__main__':
    main()

Update test_ai_agent.py