Fix order of random initial states

rl-language · Dec 4, 2024 · 3e2964d · 3e2964d
1 parent c8f29a4
commit 3e2964d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/ml/raylib/environment.py b/python/ml/raylib/environment.py
@@ -94,8 +94,8 @@ def __init__(
         self.has_score_function = has_score_function(self.module)
         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))
+        self.initial_states = random.shuffle(initial_states)
+        self.current_state = 0
         if not forced_one_player:
             self.num_agents = get_num_players(self.module)
         else: