Browse Source

allow checking if a grid contains an object defined by a pair (None, object name) (#18)

saleml 6 năm trước cách đây
mục cha
commit
341c80acd7
1 tập tin đã thay đổi với 2 bổ sung0 xóa
  1. 2 0
      gym_minigrid/minigrid.py

+ 2 - 0
gym_minigrid/minigrid.py

@@ -375,6 +375,8 @@ class Grid:
                     continue
                     continue
                 if (e.color, e.type) == key:
                 if (e.color, e.type) == key:
                     return True
                     return True
+                if key[0] is None and key[1] == e.type:
+                    return True
         return False
         return False
 
 
     def __eq__(self, other):
     def __eq__(self, other):