hot update PG

projects/codes/envs/register.py

@@ -0,0 +1,34 @@
+
+from gym.envs.registration import register
+
+def register_env(env_name):
+    if env_name == 'Racetrack-v0':
+        register(
+            id='Racetrack-v0',
+            entry_point='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()
+