Selaa lähdekoodia

Add __eq__ method for StringGymSpace - Necessary to allow for Vectorized Environments in gym

saleml 2 vuotta sitten
vanhempi
commit
c8d0c353fb
1 muutettua tiedostoa jossa 7 lisäystä ja 0 poistoa
  1. 7 0
      gym_minigrid/minigrid.py

+ 7 - 0
gym_minigrid/minigrid.py

@@ -636,6 +636,13 @@ class StringGymSpace(gym.spaces.space.Space):
     def __repr__(self):
         return "StringGymSpace(min_length={}, max_length={})".format(self.min_length, self.max_length)
 
+    def __eq__(self, other):
+        return (isinstance(other, StringGymSpace) 
+                and self.min_length == other.min_length 
+                and self.max_length == other.max_length 
+                and self.letters == other.letters
+               )
+
 class MiniGridEnv(gym.Env):
     """
     2D grid world game environment