Parcourir la source

render CartPole-v0 env

Fabian Peter Hammerle il y a 2 ans
Parent
commit
de6a8447d8
1 fichiers modifiés avec 24 ajouts et 0 suppressions
  1. 24 0
      test.py

+ 24 - 0
test.py

@@ -0,0 +1,24 @@
+#!/usr/bin/env python3
+
+import time
+
+import gym
+
+
+def _main():
+    env = gym.make("CartPole-v0")
+    env.reset()
+    env.render()
+    while True:
+        action = env.action_space.sample()
+        observation, reward, done, info = env.step(action)
+        env.render()
+        assert not info
+        print(action, observation, reward)
+        if done:
+            break
+    time.sleep(21)
+
+
+if __name__ == "__main__":
+    _main()