sorted_arguments for trace

QuantumSavory · Jun 26, 2024 · 4793dc7 · 4793dc7
1 parent 1e3c956
commit 4793dc7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/src/QSymbolicsBase/QSymbolicsBase.jl b/src/QSymbolicsBase/QSymbolicsBase.jl
@@ -1,7 +1,7 @@
 using Symbolics
 import Symbolics: simplify
 using SymbolicUtils
-import SymbolicUtils: Symbolic,_isone,flatten_term,isnotflat,Chain,Fixpoint
+import SymbolicUtils: Symbolic,_isone,flatten_term,isnotflat,Chain,Fixpoint,sorted_arguments
 using TermInterface
 import TermInterface: isexpr,head,iscall,children,operation,arguments,metadata
 

diff --git a/src/QSymbolicsBase/predefined.jl b/src/QSymbolicsBase/predefined.jl
@@ -313,6 +313,7 @@ end
 isexpr(::STrace) = true
 iscall(::STrace) = true
 arguments(x::STrace) = [x.op]
+sorted_arguments(x::STrace) = arguments(x)
 operation(x::STrace) = tr
 head(x::STrace) = :tr
 children(x::STrace) = [:tr, x.op]
@@ -322,7 +323,7 @@ tr(x::SScaled{AbstractOperator}) = x.coeff*tr(x.obj)
 tr(x::SAdd{AbstractOperator}) = (+)((tr(i) for i in arguments(x))...)
 tr(x::SOuterKetBra) = x.bra*x.ket
 tr(x::SCommutator) = 0
-tr(x::STensorOperator) = (*)((tr(i) for i in arguments(x))...) # TODO add tr properties
+tr(x::STensorOperator) = (*)((tr(i) for i in arguments(x))...)
 Base.hash(x::STrace, h::UInt) = hash((head(x), arguments(x)), h)
 Base.isequal(x::STrace, y::STrace) = isequal(x.op, y.op)