manual game works with obl pytorch

FLAIROx · Mar 5, 2024 · 768b9b2 · 768b9b2
1 parent 46153d9
commit 768b9b2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/jaxmarl/environments/hanabi/manual_game.py b/jaxmarl/environments/hanabi/manual_game.py
@@ -33,8 +33,9 @@ def act(self, env, obs, legal_moves, curr_player) -> int:
             try:
                 print("---")
                 action = int(input('Insert manual action: '))
+                print("action legal:", legal_moves[curr_player][action])
                 print("---\n")
-                if action >= 0 and action <= 20 and legal_moves[action] == 1:
+                if action >= 0 and action <= 20 and legal_moves[curr_player][action] == 1:
                     break
                 else:
                     print('Invalid action.')