34 lines
1.0 KiB
Python
34 lines
1.0 KiB
Python
|
|
from gym.envs.registration import register
|
|
|
|
def register_env(env_name):
|
|
if env_name == 'Racetrack-v0':
|
|
register(
|
|
id='Racetrack-v0',
|
|
entry_point='envs.racetrack:RacetrackEnv',
|
|
max_episode_steps=1000,
|
|
kwargs={}
|
|
)
|
|
elif env_name == 'FrozenLakeNoSlippery-v1':
|
|
register(
|
|
id='FrozenLakeNoSlippery-v1',
|
|
entry_point='gym.envs.toy_text.frozen_lake:FrozenLakeEnv',
|
|
kwargs={'map_name':"4x4",'is_slippery':False},
|
|
)
|
|
else:
|
|
print("The env name must be wrong or the environment donot need to register!")
|
|
|
|
# if __name__ == "__main__":
|
|
# import random
|
|
# import gym
|
|
# env = gym.make('FrozenLakeNoSlippery-v1')
|
|
# num_steps = 1000000
|
|
# state = env.reset()
|
|
# n_actions = env.action_space.n
|
|
# print(state)
|
|
# for _ in range(num_steps) :
|
|
# next_state, reward, done,_ = env.step(random.choice(range(n_actions)))
|
|
# print(next_state)
|
|
# if (done) :
|
|
# _ = env.reset()
|
|
|