From bacd170504f4be48635ce71390a77a6f03991b75 Mon Sep 17 00:00:00 2001
From: Marek Kaluba <kalmar@amu.edu.pl>
Date: Sun, 19 Mar 2023 20:39:43 +0100
Subject: [PATCH] preserve trace when diagonalizing M_orb

---
 src/sos_sdps.jl | 10 ++++++++--
 1 file changed, 8 insertions(+), 2 deletions(-)

diff --git a/src/sos_sdps.jl b/src/sos_sdps.jl
index eb86dcf..4eecd09 100644
--- a/src/sos_sdps.jl
+++ b/src/sos_sdps.jl
@@ -225,8 +225,14 @@ function sos_problem_primal(
         u = dot(U, iv)
 
         M_orb = invariant_constraint!(M_orb, basis(parent(elt)), cnstrs, iv)
-        Ms = SymbolicWedderburn.diagonalize!(Ms, M_orb, wedderburn)
-        SparseArrays.droptol!.(Ms, 10 * eps(T) * max(size(M_orb)...))
+
+        Ms = SymbolicWedderburn.diagonalize!(
+            Ms,
+            M_orb,
+            wedderburn;
+            trace_preserving = true,
+        )
+        # SparseArrays.droptol!.(Ms, 10 * eps(T) * max(size(M_orb)...))
 
         # @info [nnz(m) / length(m) for m in Ms]