Explorar o código

Added MiniGrid._rand_color method

Maxime Chevalier-Boisvert %!s(int64=7) %!d(string=hai) anos
pai
achega
fe469bb1cc
Modificáronse 1 ficheiros con 7 adicións e 0 borrados
  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