DO NOT SUBMIT

PiperOrigin-RevId: 384554482
google-deepmind · Jul 13, 2021 · d2a8fc5 · d2a8fc5
1 parent 6da6162
commit d2a8fc5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/distrax/__init__.py b/distrax/__init__.py
@@ -52,6 +52,7 @@
 from distrax._src.distributions.transformed import Transformed
 from distrax._src.distributions.uniform import Uniform
 
+
 # Utilities.
 from distrax._src.utils.conversion import as_bijector
 from distrax._src.utils.conversion import as_distribution

diff --git a/distrax/_src/bijectors/bijector.py b/distrax/_src/bijectors/bijector.py
@@ -94,6 +94,7 @@ def __init__(self,
         Only set to True if you're absoltely sure the Jacobian determinant is
         constant; if you're not sure, set to None.
     """
+
     if event_ndims_out is None:
       event_ndims_out = event_ndims_in
     if event_ndims_in < 0: