HF RL Explorer

Boolean indexing does not compatible with Ellipsis

Boolean indexing does not compatible with Ellipsis: a task in MiMo-V2.6-RL-harbor-code: MiMo-V2.6-RL Code (Harbor) (Harbor dataset). jnp.ones((4, 3, 2)) ..., np.array( True, False ) fail jnp.ones((4, 3, 2)) :, :, np.array( True, False ) work np.ones((4, 3, 2)) ..., np.array( True, False ) work

The task

jnp.ones((4, 3, 2))[..., np.array([True, False])] # fail jnp.ones((4, 3, 2))[:, :, np.array([True, False])] # work np.ones((4, 3, 2))[..., np.array([True, False])] # work ```

Part of FineEnvs/MiMo-V2.6-RL-harbor-code.