@classmethod
def get_observation_dims(cls, observation_shape):
    """
        Calculate observation dimension when it is flattened
        :param observation_shape:
        :return:
        """
    observation_space_size = 1
    for n in observation_shape:
        observation_space_size *= n
    return observation_space_size
