Inconsistent Dimensions for Env Observations
With the change in Obseravtion the size of observations returned by _get_observations vary. I think this is because in observations.py -> get() line 239
observation = observation + [-np.inf, -np.inf, -np.inf, -np.inf, -np.inf] * num_cells_to_fill_in
there are only 5 -np.inf beeing added, but the size of each observation should be 6 now. Changing this removed the error for me.