Fix order of random initial states

rl-language · Dec 4, 2024 · 57c596f · 57c596f
1 parent c8f29a4
commit 57c596f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/python/ml/raylib/environment.py b/python/ml/raylib/environment.py
@@ -95,7 +95,8 @@ def __init__(
         self.has_get_current_player_f = has_get_current_player(self.module)
         self.num_players = get_num_players(self.module)
         self.initial_states = initial_states
-        self.current_state = random.randint(0, len(initial_states))
+        random.shuffle(self.initial_states)
+        self.current_state = 0
         if not forced_one_player:
             self.num_agents = get_num_players(self.module)
         else: