Fix indentation

Author: leandrolcampos

tensorflow_probability/python/math/special.py

@@ -215,11 +215,11 @@ def continued_fraction_evaluation(should_stop, iteration, values, gradients):
     # We run two steps of modified Lentz's method per iteration.
     # First step of the iteration: the even one.
     new_values, new_gradients, _ = continued_fraction_step(
-      iteration, values, gradients, _betainc_even_partial_numerator)
+        iteration, values, gradients, _betainc_even_partial_numerator)
 
     # Second step of the iteration: the odd one.
     new_values, new_gradients, delta = continued_fraction_step(
-      iteration, new_values, new_gradients, _betainc_odd_partial_numerator)
+        iteration, new_values, new_gradients, _betainc_odd_partial_numerator)
 
     should_stop = should_stop | (tf.math.abs(delta - one) < tolerance)
 
@@ -420,7 +420,7 @@ def _betainc_partials(a, b, x):
   x = tf.convert_to_tensor(x, dtype=dtype)
 
   broadcast_shape = functools.reduce(
-    ps.broadcast_shape, [ps.shape(a), ps.shape(b), ps.shape(x)])
+      ps.broadcast_shape, [ps.shape(a), ps.shape(b), ps.shape(x)])
 
   a = tf.broadcast_to(a, broadcast_shape)
   b = tf.broadcast_to(b, broadcast_shape)

tensorflow_probability/python/math/special_test.py

@@ -469,8 +469,8 @@ def betainc_partials_of_partial_x(a, b, x):
 
     # Wrap in tf.function and compile for faster computations.
     betainc_partials_of_partials = [
-      tf.function(partial_fn, autograph=False, jit_compile=True)
-      for partial_fn in betainc_partials_of_partials]
+        tf.function(partial_fn, autograph=False, jit_compile=True)
+        for partial_fn in betainc_partials_of_partials]
 
     partials_of_partials = [
         partial_fn(a, b, x) for partial_fn in betainc_partials_of_partials]