gym_soccer/__init__.py (24 lines of code) (raw):

import logging from gym.envs.registration import register logger = logging.getLogger(__name__) register( id='Soccer-v0', entry_point='gym_soccer.envs:SoccerEnv', timestep_limit=1000, reward_threshold=1.0, nondeterministic = True, ) register( id='SoccerEmptyGoal-v0', entry_point='gym_soccer.envs:SoccerEmptyGoalEnv', timestep_limit=1000, reward_threshold=10.0, nondeterministic = True, ) register( id='SoccerAgainstKeeper-v0', entry_point='gym.envs:SoccerAgainstKeeperEnv', timestep_limit=1000, reward_threshold=8.0, nondeterministic = True, )