remove extra wrapping of multi_discrete space shape into a tuple (ope…

…nai#1209)
xiaoToby · Nov 6, 2018 · 6410593 · 6410593
1 parent 520fd3a
commit 6410593
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gym/spaces/multi_discrete.py b/gym/spaces/multi_discrete.py
@@ -7,7 +7,7 @@ def __init__(self, nvec):
         nvec: vector of counts of each categorical variable
         """
         self.nvec = np.asarray(nvec, dtype=np.int32)
-        gym.Space.__init__(self, (self.nvec.shape,), np.int8)
+        gym.Space.__init__(self, self.nvec.shape, np.int8)
 
     def sample(self):
         return (gym.spaces.np_random.random_sample(self.nvec.shape) * self.nvec).astype(self.dtype)