deprecated use of jnp.concat instead of jnp.concatenate in smax

FLAIROx · Mar 22, 2024 · 8615058 · 8615058
1 parent dd3bcf8
commit 8615058
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jaxmarl/environments/smax/heuristic_enemy_smax_env.py b/jaxmarl/environments/smax/heuristic_enemy_smax_env.py
@@ -98,10 +98,10 @@ def step_env(
                 self._env._decode_discrete_actions(actions)
             )
 
-        movement_actions = jnp.concat(
+        movement_actions = jnp.concatenate(
             [ally_movement_actions, enemy_movement_actions], axis=0
         )
-        attack_actions = jnp.concat([ally_attack_actions, enemy_attack_actions], axis=0)
+        attack_actions = jnp.concatenate([ally_attack_actions, enemy_attack_actions], axis=0)
 
         if not get_state_sequence:
             obs, jaxmarl_state, rewards, dones, infos = self._env.step_env_no_decode(