'Sample single step from policy.'

def sample_step(self, obs, prev_internal_state, prev_actions, greedy=False):

    (next_state, sampled_actions, logits, log_probs, entropies, self_kls) = self.single_step((prev_internal_state, prev_actions, None, None, None, None), (obs, None), greedy=greedy)
    return (next_state, sampled_actions)
