Browse Source

Added MiniGrid._rand_color method

Maxime Chevalier-Boisvert 7 năm trước cách đây
mục cha
commit
fe469bb1cc
1 tập tin đã thay đổi với 7 bổ sung0 xóa
  1. 7 0
      gym_minigrid/minigrid.py

+ 7 - 0
gym_minigrid/minigrid.py

@@ -866,6 +866,13 @@ class MiniGridEnv(gym.Env):
         idx = self._rand_int(0, len(lst))
         return lst[idx]
 
+    def _rand_color(self):
+        """
+        Generate a random color name (string)
+        """
+
+        return self._rand_elem(COLOR_NAMES)
+
     def _rand_pos(self, xLow, xHigh, yLow, yHigh):
         """
         Generate a random (x,y) position tuple