Skip to content

Commit

Permalink
pep8
Browse files Browse the repository at this point in the history
  • Loading branch information
rflamary committed Oct 30, 2023
1 parent de19529 commit 418a1de
Showing 1 changed file with 4 additions and 3 deletions.
7 changes: 4 additions & 3 deletions ot/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -567,7 +567,7 @@ def reduce_lazytensor(a, func, axis=None, nx=None, batch_size=100):

def get_lowrank_lazytensor(Q, R, d=None, nx=None):
""" Get a low rank LazyTensor T=Q@R^T or T=Q@diag(d)@R^T
Parameters
----------
Q : ndarray, shape (n, r)
Expand All @@ -578,11 +578,12 @@ def get_lowrank_lazytensor(Q, R, d=None, nx=None):
Diagonal of the lowrank tensor
nx : Backend, optional
Backend to use for the reduction
Returns
-------
T : LazyTensor
Lowrank tensor T=Q@R^T or T=Q@diag(d)@R^T """
Lowrank tensor T=Q@R^T or T=Q@diag(d)@R^T
"""

if nx is None:
nx = get_backend(Q, R, d)
Expand Down

0 comments on commit 418a1de

Please sign in to comment.