docs: def sample_given_context(self, c, c_dims):
          '''
          Sample the region with max progress among regions that have the same context
              c: context value on c_dims dimensions
              c_dims: w.r.t sensory space dimensions
          '''
          index = self.discrete_progress.sample_given_context(c, c_dims, self.space)
          return self.space.rand_value(index).flatten()[list(set(range(len(self.space.cardinalities))) - set(c_dims))]