def _get_observation_space_(self) -> D.T_agent[Space[D.T_observation]]: return MultiDiscreteSpace( [self._n_positions + 1, self._n_positions + 1])
def _get_observation_space_(self) -> D.T_agent[Space[D.T_observation]]: return MultiDiscreteSpace([self.num_cols, self.num_rows, 100])
def _get_observation_space_(self) -> D.T_agent[Space[D.T_observation]]: return StrDict[MultiDiscreteSpace([self._num_cols, self.n_um_rows])]