JET fixes

QuantumSavory · Jun 27, 2024 · 343108b · 343108b
1 parent 871b1c3
commit 343108b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/QSymbolicsBase/predefined_CPTP.jl b/src/QSymbolicsBase/predefined_CPTP.jl
@@ -73,10 +73,10 @@ julia> K*ρ
 end
 isexpr(::KrausRepr) = true
 iscall(::KrausRepr) = true
-arguments(x::KrausRepr) = [sop, x.krausops]
+arguments(x::KrausRepr) = [x.sop, x.krausops]
 operation(x::KrausRepr) = kraus
 head(x::KrausRepr) = :kraus
-children(x::KrausRepr) = [:kraus, sop, x.krausops]
+children(x::KrausRepr) = [:kraus, x.sop, x.krausops]
 kraus(s::Symbolic{AbstractSuperOperator}, xs::Symbolic{AbstractOperator}...) = KrausRepr(s,collect(xs))
 symbollabel(x::KrausRepr) = symbollabel(x.sop)
 basis(x::KrausRepr) = basis(x.sop)

diff --git a/src/QSymbolicsBase/rules.jl b/src/QSymbolicsBase/rules.jl
@@ -63,7 +63,7 @@ end
 # Flattening expressions
 RULES_FLATTEN = [
     @rule(~x::isnotflat_precheck(⊗) => flatten_term(⊗, ~x)),
-    @rule ⊗(~~xs::hasscalings) => prefactorscalings_rule(xs)
+    @rule ⊗(~~xs::hasscalings) => prefactorscalings_rule(~~xs)
 ]
 
 # Pauli identities