def convert_obs_to_list(self, obs):
    if (len(self.obs_dims) == 1):
        return [obs]
    else:
        return list(obs)
def convert_obs_to_list(self, obs):
    if (len(self.obs_dims) == 1):
        return [obs]
    else:
        return list(obs)