Skip to content

Commit db459b5

Browse files
fix: retain Num type on scalarize
1 parent 3875b9d commit db459b5

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

src/Symbolics.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -219,18 +219,18 @@ for T in [Num, Complex{Num}]
219219
@eval begin
220220
#(::Type{S})(x::$T) where {S<:Union{NumberTypes,AbstractArray}} = S(Symbolics.unwrap(x))::S
221221

222-
SymbolicUtils.simplify(n::$T; kw...) = wrap(SymbolicUtils.simplify(unwrap(n); kw...))
223-
SymbolicUtils.simplify_fractions(n::$T; kw...) = wrap(SymbolicUtils.simplify_fractions(unwrap(n); kw...))
224-
SymbolicUtils.expand(n::$T) = wrap(SymbolicUtils.expand(unwrap(n)))
222+
SymbolicUtils.simplify(n::$T; kw...) = $T(SymbolicUtils.simplify(unwrap(n); kw...))
223+
SymbolicUtils.simplify_fractions(n::$T; kw...) = $T(SymbolicUtils.simplify_fractions(unwrap(n); kw...))
224+
SymbolicUtils.expand(n::$T) = $T(SymbolicUtils.expand(unwrap(n)))
225225

226226
SymbolicUtils.Code.toexpr(x::$T) = SymbolicUtils.Code.toexpr(unwrap(x))
227227

228-
SymbolicUtils.setmetadata(x::$T, t, v) = wrap(SymbolicUtils.setmetadata(unwrap(x), t, v))
228+
SymbolicUtils.setmetadata(x::$T, t, v) = $T(SymbolicUtils.setmetadata(unwrap(x), t, v))
229229
SymbolicUtils.getmetadata(x::$T, t) = SymbolicUtils.getmetadata(unwrap(x), t)
230230
SymbolicUtils.hasmetadata(x::$T, t) = SymbolicUtils.hasmetadata(unwrap(x), t)
231231

232232
Broadcast.broadcastable(x::$T) = x
233-
SymbolicUtils.scalarize(x::$T) = scalarize(unwrap(x))
233+
SymbolicUtils.scalarize(x::$T) = $T(scalarize(unwrap(x)))
234234
end
235235
end
236236

0 commit comments

Comments
 (0)