diff --git a/pyro/distributions/torch_distribution.py b/pyro/distributions/torch_distribution.py index ace02da72a..379dd8f01a 100644 --- a/pyro/distributions/torch_distribution.py +++ b/pyro/distributions/torch_distribution.py @@ -3,7 +3,6 @@ import warnings from collections import OrderedDict -from typing import Callable import torch from torch.distributions.kl import kl_divergence, register_kl @@ -16,7 +15,7 @@ from .util import broadcast_shape, scale_and_mask -class TorchDistributionMixin(Distribution, Callable): +class TorchDistributionMixin(Distribution): """ Mixin to provide Pyro compatibility for PyTorch distributions.