a1c1bdc135a1f030a5c306452c02242669e405d4,mushroom/policy/torch_policy.py,TorchPolicy,distribution,#TorchPolicy#Any#,39

Before Change

        return torch.squeeze(a, dim=0).detach().numpy()

    def distribution(self, state):
        s = torch.tensor(state, dtype=torch.float)
        return self.distribution_t(s)

    def entropy(self, state=None):
        s = torch.tensor(state, dtype=torch.float) if state is not None else None

After Change

        return torch.squeeze(a, dim=0).detach().cpu().numpy()

    def distribution(self, state):
        s = self._to_tensor(state)
        return self.distribution_t(s)

    def entropy(self, state=None):
        s = self._to_tensor(state) if state is not None else None

Instances