Fix D instantiation

JuliaSymbolics · Feb 19, 2024 · 0a90248 · 0a90248
1 parent b7f8b5c
commit 0a90248
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/diff.jl b/src/diff.jl
@@ -37,12 +37,12 @@ end
 function (D::Differential)(x)
     x = unwrap(x)
     if isarraysymbolic(x)
-        wrap(array_term(D, x))
+        array_term(D, x)
     else
-        wrap(term(D, x))
+        term(D, x)
     end
 end
-(D::Differential)(x::Num) = Num(D(value(x)))
+(D::Differential)(x::Union{Num, Arr}) = wrap(D(unwrap(x)))
 (D::Differential)(x::Complex{Num}) = wrap(ComplexTerm{Real}(D(unwrap(real(x))), D(unwrap(imag(x)))))
 SymbolicUtils.promote_symtype(::Differential, T) = T