Add Dict helper methods (openai#2417)

FragileTech · Sep 22, 2021 · ae01834 · ae01834
1 parent d35d211
commit ae01834
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/gym/spaces/dict.py b/gym/spaces/dict.py
@@ -147,3 +147,12 @@ def from_jsonable(self, sample_n):
 
     def __eq__(self, other):
         return isinstance(other, Dict) and self.spaces == other.spaces
+
+    def keys(self):
+        return self.spaces.keys()
+
+    def values(self):
+        return self.spaces.values()
+
+    def items(self):
+        return self.spaces.items()