Update distance.py

pyRiemann · Dec 17, 2024 · 9232b73 · 9232b73
1 parent 95c0ed1
commit 9232b73
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyriemann_qiskit/utils/distance.py b/pyriemann_qiskit/utils/distance.py
@@ -111,7 +111,7 @@ def weights_logeuclid_to_convex_hull(A, B, optimizer=ClassicalOptimizer()):
     matrices = range(n_matrices)
 
     def log_prod(m1, m2):
-        return np.nansum(logm(m1).flatten() * logm(m2).flatten())
+        return np.trace(logm(m1) @ logm(m2))
 
     prob = Model()
     optimizer = get_global_optimizer(optimizer)