Update distance.py

pyRiemann · Dec 20, 2024 · bd54e88 · bd54e88
1 parent 425c99d
commit bd54e88
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyriemann_qiskit/utils/distance.py b/pyriemann_qiskit/utils/distance.py
@@ -118,9 +118,9 @@ def trace_prod_log(m1, m2):
     w = optimizer.get_weights(prob, matrices)
 
     wtLogAtLogAw = prob.sum(
-        w[i] * w[j] * log_prod(A[i], A[j]) for i in matrices for j in matrices
+        w[i] * w[j] * trace_prod_log(A[i], A[j]) for i in matrices for j in matrices
     )
-    wLogBLogA = prob.sum(w[i] * log_prod(B, A[i]) for i in matrices)
+    wLogBLogA = prob.sum(w[i] * trace_prod_log(B, A[i]) for i in matrices)
     objective = wtLogAtLogAw - 2 * wLogBLogA
 
     prob.set_objective("min", objective)