do version 0.9.0 Pt.2

Techtonique · Aug 10, 2023 · 94d11fe · 94d11fe
1 parent 243e660
commit 94d11fe
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/teller/utils/numerical_gradient.py b/teller/utils/numerical_gradient.py
@@ -38,7 +38,7 @@ def numerical_gradient(f, X, normalize=False, h=None, n_jobs=None, verbose=1):
 
                 X[:, ix] = value_x  # restore (!)
 
-                grad[:, ix] = (fx_plus - fx_minus) / double_h
+                grad[:, ix] = (np.asarray(fx_plus) - np.asarray(fx_minus)) / double_h
 
                 pbar.update(ix)
 
@@ -75,7 +75,7 @@ def numerical_gradient(f, X, normalize=False, h=None, n_jobs=None, verbose=1):
 
             X[:, ix] = value_x  # restore (!)
 
-            grad[:, ix] = (fx_plus - fx_minus) / (2 * h)
+            grad[:, ix] = (np.asarray(fx_plus) - np.asarray(fx_minus)) / (2 * h)
 
             if verbose == 1:
                 pbar.update(ix)
@@ -107,7 +107,7 @@ def gradient_column(ix):
 
         X[:, ix] = value_x  # restore (!)
 
-        grad[:, ix] = (fx_plus - fx_minus) / (2 * h)
+        grad[:, ix] = (np.asarray(fx_plus) - np.asarray(fx_minus)) / (2 * h)
 
     if verbose == 1:
         print("\n")