def get_actions(self, observations):
    'Return multiple actions.'
    flat_obs = self.observation_space.flatten_n(observations)
    probs = self._f_prob(flat_obs)
    actions = list(map(self.action_space.weighted_sample, probs))
    return (actions, dict(prob=probs))
Return multiple actions.
def get_actions(self, observations):
    
    flat_obs = self.observation_space.flatten_n(observations)
    probs = self._f_prob(flat_obs)
    actions = list(map(self.action_space.weighted_sample, probs))
    return (actions, dict(prob=probs))