added minimal_action_set function to each environment

kenjyoung · Mar 25, 2019 · ce4706e · ce4706e
1 parent 2858af7
commit ce4706e
Show file tree

Hide file tree

Showing 7 changed files with 24 additions and 1 deletion.
diff --git a/environments/asterix.py b/environments/asterix.py
@@ -149,3 +149,7 @@ def reset(self):
     # Dimensionality of the game-state (10x10xn)
     def state_shape(self):
         return [10,10,len(self.channels)]
+
+    def minimal_action_set(self):
+        minimal_actions = ['n','l','u','r','d']
+        return [self.action_map.index(x) for x in minimal_actions]
diff --git a/environments/breakout.py b/environments/breakout.py
@@ -124,3 +124,7 @@ def reset(self):
     # Dimensionality of the game-state (10x10xn)
     def state_shape(self):
         return [10,10,len(self.channels)]
+
+    def minimal_action_set(self):
+        minimal_actions = ['n','l','r']
+        return [self.action_map.index(x) for x in minimal_actions]
diff --git a/environments/freeway.py b/environments/freeway.py
@@ -138,3 +138,7 @@ def reset(self):
     # Dimensionality of the game-state (10x10xn)
     def state_shape(self):
         return [10,10,len(self.channels)]
+
+    def minimal_action_set(self):
+        minimal_actions = ['n','u','d']
+        return [self.action_map.index(x) for x in minimal_actions]
diff --git a/environments/seaquest.py b/environments/seaquest.py
@@ -305,3 +305,7 @@ def reset(self):
     # Dimensionality of the game-state (10x10xn)
     def state_shape(self):
         return [10,10,len(self.channels)]
+
+    def minimal_action_set(self):
+        minimal_actions = ['n','l','u','r','d','f']
+        return [self.action_map.index(x) for x in minimal_actions]
diff --git a/environments/space_invaders.py b/environments/space_invaders.py
@@ -150,3 +150,7 @@ def reset(self):
     # Dimensionality of the game-state (10x10xn)
     def state_shape(self):
         return [10,10,len(self.channels)]
+
+    def minimal_action_set(self):
+        minimal_actions = ['n','l','r','f']
+        return [self.action_map.index(x) for x in minimal_actions]
diff --git a/minatar_environment/environment.py b/minatar_environment/environment.py
@@ -48,3 +48,6 @@ def num_actions(self):
 
     def game_name(self):
         return self.env_name
+
+    def minimal_action_set(self):
+        return self.env.minimal_action_set()
diff --git a/setup.py b/setup.py
@@ -1,7 +1,7 @@
 from distutils.core import setup
 
 setup(name='MinAtar',
-      version='1.0.1',
+      version='1.0.2',
       description='A miniaturized version of the arcade learning environment.',
       url='https://github.com/kenjyoung/MinAtar',
       author='Kenny Young',