e.g.
size(reset(antenv), 1) == antenv.observation_space.n
returns false.
I'll add those tests in a PR.
Reason is because although wedefine observation space as https://github.com/dojo-sim/Dojo.jl/blob/main/environments/ant/methods/env.jl#L44-L51 but then get_observation concatenates the contact forces as well: https://github.com/dojo-sim/Dojo.jl/blob/main/environments/ant/methods/env.jl#L175. We need a way to reflect this in the observation space.