From 69503dafb13a5ba3def86508254aba7d755aca28 Mon Sep 17 00:00:00 2001 From: Brandon Rozek Date: Mon, 27 Jan 2020 21:16:55 -0500 Subject: [PATCH] Preparing gymrecord to be snappified --- play_env.py => gymrecord | 3 ++- play_pong.sh | 2 +- requirements.txt | 16 ++++++++++++++++ setup.py | 6 ++++++ 4 files changed, 25 insertions(+), 2 deletions(-) rename play_env.py => gymrecord (98%) create mode 100644 requirements.txt create mode 100644 setup.py 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'] +)