diff options
author | RĂ©mi Flamary <remi.flamary@gmail.com> | 2023-04-07 16:24:03 +0200 |
---|---|---|
committer | GitHub <noreply@github.com> | 2023-04-07 16:24:03 +0200 |
commit | 2bbfbbb18cf0a0544f0913f1917d08fb5384ac9b (patch) | |
tree | 69007360687aa282438c44bd3fc060981d492870 /ot | |
parent | 89f1613861152432807077fbb146578611dc5888 (diff) |
[MRG] debug documentation (#453)
Diffstat (limited to 'ot')
-rw-r--r-- | ot/optim.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/ot/optim.py b/ot/optim.py index 58e5596..b15c77b 100644 --- a/ot/optim.py +++ b/ot/optim.py @@ -93,7 +93,7 @@ def line_search_armijo( fc[0] += 1 alpha10 = nx.from_numpy(alpha1) fval = f(xk0 + alpha10 * pk0, *args) - if type(fval) is float: + if isinstance(fval, float): # prevent bug from nx.to_numpy that can look for .cpu or .gpu return fval else: @@ -101,7 +101,7 @@ def line_search_armijo( if old_fval is None: phi0 = phi(0.) - elif type(old_fval) is float: + elif isinstance(old_fval, float): # prevent bug from nx.to_numpy that can look for .cpu or .gpu phi0 = old_fval else: |