Add missing super().__init__()

pyro-ppl · Jan 21, 2021 · 29d61ef · 29d61ef
1 parent 9153c13
commit 29d61ef
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/numpyro/distributions/flows.py b/numpyro/distributions/flows.py
@@ -37,6 +37,7 @@ def __init__(self, autoregressive_nn, log_scale_min_clip=-5., log_scale_max_clip
         :param autoregressive_nn: an autoregressive neural network whose forward call returns a real-valued
             mean and log scale as a tuple
         """
+        super().__init__()
         self.arn = autoregressive_nn
         self.log_scale_min_clip = log_scale_min_clip
         self.log_scale_max_clip = log_scale_max_clip
@@ -96,6 +97,7 @@ class BlockNeuralAutoregressiveTransform(Transform):
     event_dim = 1
 
     def __init__(self, bn_arn):
+        super().__init__()
         self.bn_arn = bn_arn
 
     def __call__(self, x):