def dist_info(self, obs, state_infos=None):
    'Distribution info.'
    prob = self._f_prob(obs)
    return dict(prob=prob)
Distribution info.
def dist_info(self, obs, state_infos=None):
    
    prob = self._f_prob(obs)
    return dict(prob=prob)