new gym spaces API with dtype; updated required packages

zuoxingdong · Mar 19, 2018 · 7033a1b · 7033a1b
1 parent 8aaeb4d
commit 7033a1b
Show file tree

Hide file tree

Showing 4 changed files with 28 additions and 12 deletions.
diff --git a/Maze.ipynb b/Maze.ipynb
@@ -1227,7 +1227,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.6.3"
+   "version": "3.6.4"
   }
  },
  "nbformat": 4,

diff --git a/README.md b/README.md
@@ -9,12 +9,12 @@ The repo will be actively maintained, any comments, feedbacks or improvements ar
 # Dependencies
 
 This repository requires following packages:
-- gym >= 0.9.4
-- Python >= 3.6
-- Numpy >= 1.13.3
-- Matplotlib >= 2.1.1
+- gym >= 0.10.4
+- Python >= 3.6.4
+- Numpy >= 1.14.2
+- Matplotlib >= 2.1.2
 - Scikit-image >= 0.13.1
-- (Recommended) JupyterLab >= 0.31.0
+- (Recommended) JupyterLab >= 0.31.12
 
 # Installation
 

diff --git a/gym_maze/envs/maze.py b/gym_maze/envs/maze.py
@@ -55,11 +55,15 @@ def __init__(self,
         low_obs = 0  # Lowest integer in observation
         high_obs = 6  # Highest integer in observation
         if self.obs_type == 'full':
-            self.observation_space = spaces.Box(low=low_obs, high=high_obs,
-                                                shape=self.maze_size)
+            self.observation_space = spaces.Box(low=low_obs, 
+                                                high=high_obs,
+                                                shape=self.maze_size, 
+                                                dtype=np.float32)
         elif self.obs_type == 'partial':
-            self.observation_space = spaces.Box(low=low_obs, high=high_obs,
-                                                shape=(self.pob_size*2+1, self.pob_size*2+1))
+            self.observation_space = spaces.Box(low=low_obs, 
+                                                high=high_obs,
+                                                shape=(self.pob_size*2+1, self.pob_size*2+1), 
+                                                dtype=np.float32)
         else:
             raise TypeError('Observation type must be either \'full\' or \'partial\'')
 

diff --git a/setup.py b/setup.py
@@ -1,6 +1,18 @@
 from setuptools import setup
+import sys
+
+# Only support Python 3
+if sys.version_info.major != 3:
+    print(f'WARNING: This package only officially support Python 3, the current version is Python {sys.version_info.major}. The installation will likely fail. ')
 
 setup(name='gym_maze',
-      version='1.0',
-      install_requires=['gym', 'numpy', 'scikit-image']
+      install_requires=['gym', 
+                        'numpy', 
+                        'matplotlib', 
+                        'scikit-image', 
+                        'jupyterlab'],
+      description='gym-maze: A customizable gym environment for maze and gridworld',
+      author='Xingdong Zuo',
+      url='https://github.com/zuoxingdong/gym-maze',
+      version='0.1'
 )