928f1d0b3ba401ea7614f19f3595e1a44c171205,maml_rl/utils/torch_utils.py,,detach_distribution,#Any#,36

Before Change

    elif isinstance(pi, Normal):
        distribution = Normal(loc=pi.loc.detach(), scale=pi.scale.detach())
    else:
        raise NotImplementedError("Only `Categorical` and `Normal` "
                                  "policies are valid policies.")
    return distribution

def to_numpy(tensor):
    if isinstance(tensor, torch.Tensor):

After Change

    return out

def detach_distribution(pi):
    if isinstance(pi, Independent):
        distribution = Independent(detach_distribution(pi.base_dist),
                                   pi.reinterpreted_batch_ndims)
    elif isinstance(pi, Categorical):
        distribution = Categorical(logits=pi.logits.detach())
    elif isinstance(pi, Normal):
        distribution = Normal(loc=pi.loc.detach(), scale=pi.scale.detach())
    else:
        raise NotImplementedError("Only `Categorical`, `Independent` and "
                                  "`Normal` policies are valid policies. Got "
                                  "`{0}`.".format(type(pi)))
    return distribution

def to_numpy(tensor):
    if isinstance(tensor, torch.Tensor):

Instances