diff --git a/src/relaxit/distributions/GaussianRelaxedBernoulli.py b/src/relaxit/distributions/GaussianRelaxedBernoulli.py index af1ae35..9dc719d 100644 --- a/src/relaxit/distributions/GaussianRelaxedBernoulli.py +++ b/src/relaxit/distributions/GaussianRelaxedBernoulli.py @@ -1,3 +1,14 @@ +''' +The :mod:`relaxit.distributions.GaussianRelaxedBernoulli` contains classes: + +- :class:`relaxit.distributions.GaussianRelaxedBernoulli.GaussianRelaxedBernoulli` + +''' +from __future__ import print_function + +__docformat__ = 'restructuredtext' + + import torch from pyro.distributions.torch_distribution import TorchDistribution from torch.distributions import constraints