5537fe13b097097668f9c08a00051e8b7a2d1980,rllib/utils/exploration/random.py,Random,get_torch_exploration_action,#Random#Any#Any#,83

Before Change

        if explore:
            // Unsqueeze will be unnecessary, once we support batch/time-aware
            // Spaces.
            action = tensor_fn(self.action_space.sample()).unsqueeze(0)
        else:
            action = tensor_fn(action_dist.deterministic_sample())
        logp = torch.zeros((action.size()[0], ), dtype=torch.float32)
        return action, logp

After Change

            // Unsqueeze will be unnecessary, once we support batch/time-aware
            // Spaces.
            a = self.action_space.sample()
            action = tensor_fn([a] if isinstance(a, int) else a)
        else:
            action = tensor_fn(action_dist.deterministic_sample())
        logp = torch.zeros((action.size()[0], ), dtype=torch.float32)

Instances