Fix docs

pyro-ppl · Oct 2, 2019 · b68c286 · b68c286
1 parent e881c24
commit b68c286
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyro/distributions/transforms/batch_norm.py b/pyro/distributions/transforms/batch_norm.py
@@ -34,7 +34,7 @@ class BatchNormTransform(TransformModule):
     Example usage:
 
     >>> from pyro.nn import AutoRegressiveNN
-    >>> from pyro.distributions import InverseAutoregressiveFlow
+    >>> from pyro.distributions.transforms import InverseAutoregressiveFlow
     >>> base_dist = dist.Normal(torch.zeros(10), torch.ones(10))
     >>> iafs = [InverseAutoregressiveFlow(AutoRegressiveNN(10, [40])) for _ in range(2)]
     >>> bn = BatchNormTransform(10)

diff --git a/pyro/distributions/transforms/permute.py b/pyro/distributions/transforms/permute.py
@@ -20,7 +20,7 @@ class PermuteTransform(Transform):
     Example usage:
 
     >>> from pyro.nn import AutoRegressiveNN
-    >>> from pyro.distributions import InverseAutoregressiveFlow, PermuteTransform
+    >>> from pyro.distributions.transforms import InverseAutoregressiveFlow, PermuteTransform
     >>> base_dist = dist.Normal(torch.zeros(10), torch.ones(10))
     >>> iaf1 = InverseAutoregressiveFlow(AutoRegressiveNN(10, [40]))
     >>> ff = PermuteTransform(torch.randperm(10, dtype=torch.long))