Change deck size type.

FLAIROx · Aug 3, 2024 · 1e8587c · 1e8587c
1 parent 710e292
commit 1e8587c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jaxmarl/environments/hanabi/hanabi_game.py b/jaxmarl/environments/hanabi/hanabi_game.py
@@ -59,7 +59,7 @@ def __init__(
         self.max_info_tokens = max_info_tokens
         self.max_life_tokens = max_life_tokens
         self.num_cards_of_rank = num_cards_of_rank
-        self.deck_size = jnp.sum(num_cards_of_rank) * num_colors
+        self.deck_size = np.sum(num_cards_of_rank) * num_colors
         self.color_map = color_map
 
         # action ranges - useful to know