Skip to content

Commit 8f45820

Browse files
oskarfernlundjburnim
authored andcommitted
pylint errors
1 parent 3871387 commit 8f45820

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tensorflow_probability/python/math/linalg.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -294,8 +294,8 @@ def pivoted_cholesky(matrix,
294294
diag_rtol: Scalar floating point `Tensor` (same dtype as `matrix`). If the
295295
errors of all diagonal elements of `lr @ lr.T` are each lower than
296296
`element * diag_rtol`, iteration is permitted to terminate early.
297-
return_pivoting_order: If `True`, return an `int` `Tensor` indicating the pivoting
298-
order used to produce `lr` in addition to `lr` (defaults to `False`).
297+
return_pivoting_order: If `True`, return an `int` `Tensor` indicating the
298+
pivoting order used to produce `lr` (in addition to `lr`).
299299
name: Optional name for the op.
300300
301301
Returns:
@@ -412,7 +412,7 @@ def body(m, pchol, perm, matrix_diag):
412412
pchol = tf.linalg.matrix_transpose(pchol)
413413
tensorshape_util.set_shape(
414414
pchol, tensorshape_util.concatenate(matrix_diag.shape, [None]))
415-
415+
416416
if return_pivoting_order:
417417
return pchol, perm
418418
else:

0 commit comments

Comments
 (0)