diff options
-rw-r--r-- | docs/source/conf.py | 3 | ||||
-rw-r--r-- | ot/optim.py | 4 |
2 files changed, 4 insertions, 3 deletions
diff --git a/docs/source/conf.py b/docs/source/conf.py index 6e76291..b7a9e35 100644 --- a/docs/source/conf.py +++ b/docs/source/conf.py @@ -75,7 +75,8 @@ extensions = [ 'sphinx.ext.viewcode', 'sphinx.ext.napoleon', 'sphinx_gallery.gen_gallery', - 'myst_parser' + 'myst_parser', + "sphinxcontrib.jquery", ] autosummary_generate = True 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: |