Make batch_shape and event_shape required

pyro-ppl · Jun 2, 2020 · 4030b96 · 4030b96
1 parent 244eb58
commit 4030b96
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/pyro/distributions/improper_uniform.py b/pyro/distributions/improper_uniform.py
@@ -37,12 +37,12 @@ class ImproperUniform(TorchDistribution):
 
     :param support: The support of the distribution.
     :type support: ~torch.distributions.constraints.Constraint
-    :param torch.Size batch_shape: Defaults to empty.
-    :param torch.Size event_shape: Defaults to empty.
+    :param torch.Size batch_shape: The batch shape.
+    :param torch.Size event_shape: The event shape.
     """
     arg_constraints = {}
 
-    def __init__(self, support, batch_shape=torch.Size(), event_shape=torch.Size()):
+    def __init__(self, support, batch_shape, event_shape):
         assert isinstance(support, constraints.Constraint)
         self._support = support
         super().__init__(batch_shape, event_shape)