From 2bbfbbb18cf0a0544f0913f1917d08fb5384ac9b Mon Sep 17 00:00:00 2001
From: Rémi Flamary <remi.flamary@gmail.com>
Date: Fri, 7 Apr 2023 16:24:03 +0200
Subject: [MRG] debug documentation (#453)

---
 ot/optim.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

(limited to 'ot')

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:
-- 
cgit v1.2.3