forked from openai/gym
-
Notifications
You must be signed in to change notification settings - Fork 0
/
check_envs_for_change.py
37 lines (28 loc) · 1.39 KB
/
check_envs_for_change.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
ENVS = ["Ant-v0", "HalfCheetah-v0", "Hopper-v0", "Humanoid-v0", "InvertedDoublePendulum-v0", "Reacher-v0", "Swimmer-v0", "Walker2d-v0"]
OLD_COMMIT = "HEAD"
# ================================================================
import subprocess, gym
from gym import utils
from os import path
def cap(cmd):
"Call and print command"
print utils.colorize(cmd, "green")
subprocess.check_call(cmd,shell=True)
# ================================================================
gymroot = path.abspath(path.dirname(path.dirname(gym.__file__)))
oldgymroot = "/tmp/old-gym"
comparedir = "/tmp/gym-comparison"
oldgymbase = path.basename(oldgymroot)
print "gym root", gymroot
thisdir = path.abspath(path.dirname(__file__))
print "this directory", thisdir
cap("rm -rf %(oldgymroot)s %(comparedir)s && mkdir %(comparedir)s && cd /tmp && git clone %(gymroot)s %(oldgymbase)s"%locals())
for env in ENVS:
print utils.colorize("*"*50 + "\nENV: %s" % env, "red")
writescript = path.join(thisdir, "write_rollout_data.py")
outfileA = path.join(comparedir, env) + "-A.npz"
cap("python %(writescript)s %(env)s %(outfileA)s"%locals())
outfileB = path.join(comparedir, env) + "-B.npz"
cap("python %(writescript)s %(env)s %(outfileB)s --gymdir=%(oldgymroot)s"%locals())
comparescript = path.join(thisdir, "compare_rollout_data.py")
cap("python %(comparescript)s %(outfileA)s %(outfileB)s"%locals())