bugfixes

jdtuck · Dec 19, 2023 · fcebf9b · fcebf9b
1 parent 7755d85
commit fcebf9b
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/fdasrsf/umap_metric.py b/fdasrsf/umap_metric.py
@@ -268,7 +268,9 @@ def warp_rbfgs(q1, q2):
 
     hCurGradNorm = norm2(hCurGradient)
 
-    lsstats = {"costevals": 0, "stepsize": 0.0, "alpha": 0.0}
+    costevals = 0
+    stepsize = 0.0
+    alpha = 0.0
 
     ultimatum = False
 
@@ -308,6 +310,9 @@ def warp_rbfgs(q1, q2):
         htilde = group_action_SRVF(htilde, hNext)
         q2tilde = group_action_SRVF(q2tilde, hNext)
 
+        stepsize = lsstats['stepsize']
+        alpha = lsstats['alpha']
+
         alpha = stepsize / norm2(p)
         step = alpha * p