diff options
| author | Nathan Cassereau <84033440+ncassereau-idris@users.noreply.github.com> | 2021-12-03 12:37:05 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2021-12-03 12:37:05 +0100 |
| commit | ca69658400dc2ef6a7d3e531acffcd107400085f (patch) | |
| tree | b77a28821067be5240cec2082fa1f119b1cfd1cd /ot/optim.py | |
| parent | cb510644b2fd65e4ce216a7799ce7401f71548b8 (diff) | |
[MRG] Cupy backend (#315)
* Cupy backend
* pep8 + bug
* working even if cupy not installed
* attempt to force codecov to ignore cupy because no gpu can be used for testing on github
* docstring
* readme
Diffstat (limited to 'ot/optim.py')
| -rw-r--r-- | ot/optim.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/ot/optim.py b/ot/optim.py index 9b8a8f7..f25e2c9 100644 --- a/ot/optim.py +++ b/ot/optim.py @@ -88,7 +88,7 @@ def line_search_armijo( else: if alpha_min is not None or alpha_max is not None: alpha = np.clip(alpha, alpha_min, alpha_max) - return alpha, fc[0], phi1 + return float(alpha), fc[0], phi1 def solve_linesearch( |