Переглянути джерело

Added MiniGrid._rand_color method

Maxime Chevalier-Boisvert 7 роки тому
батько
коміт
fe469bb1cc
1 змінених файлів з 7 додано та 0 видалено
  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