add __setitem__ method to Dict action space (#2377)

openai · Sep 1, 2021 · cf0cf17 · cf0cf17
1 parent a004558
commit cf0cf17
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gym/spaces/dict.py b/gym/spaces/dict.py
@@ -70,6 +70,9 @@ def contains(self, x):
     def __getitem__(self, key):
         return self.spaces[key]
 
+    def __setitem__(self, key, value):
+        self.spaces[key] = value
+
     def __iter__(self):
         for key in self.spaces:
             yield key