diff --git a/play_env.py b/gymrecord similarity index 98% rename from play_env.py rename to gymrecord index 9b99d4d..86aaefa 100644 --- a/play_env.py +++ b/gymrecord @@ -1,3 +1,4 @@ +#!/usr/bin/env python import play import gym from collections import namedtuple @@ -91,4 +92,4 @@ while playThread.is_alive(): env.log_transitions() # Save what's remaining after process died -env.log_transitions() \ No newline at end of file +env.log_transitions() diff --git a/play_pong.sh b/play_pong.sh index b7591ce..9b79ca8 100755 --- a/play_pong.sh +++ b/play_pong.sh @@ -1,2 +1,2 @@ #!/bin/sh -python play_env.py --environment_name=PongNoFrameskip-v4 --logdir=playlogs +python gymrecord --environment_name=PongNoFrameskip-v4 --logdir=playlogs diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..044ab6d --- /dev/null +++ b/requirements.txt @@ -0,0 +1,16 @@ +atari-py==0.2.6 +cloudpickle==1.2.2 +cycler==0.10.0 +future==0.18.2 +gym==0.15.4 +kiwisolver==1.1.0 +matplotlib==3.1.2 +numpy==1.18.1 +opencv-python==4.1.2.30 +Pillow==7.0.0 +pygame==1.9.6 +pyglet==1.3.2 +pyparsing==2.4.6 +python-dateutil==2.8.1 +scipy==1.4.1 +six==1.14.0 diff --git a/setup.py b/setup.py new file mode 100644 index 0000000..c3b4075 --- /dev/null +++ b/setup.py @@ -0,0 +1,6 @@ +from distutils.core import setup +setup(name='gymrecord', + version='0.9', + scripts=['gymrecord'], + py_modules = ['play'] +)