feat: also use pretty printing for regular show

Author: MilesCranmer

src/ExtensionInterface.jl

@@ -23,6 +23,7 @@ function Base.show(io::IO, g::ZygoteGradient{F,degree,arg}) where {F,degree,arg}
     print(io, g.op)
     return nothing
 end
+Base.show(io::IO, ::MIME"text/plain", g::ZygoteGradient) = show(io, g)
 
 function _zygote_gradient(args...)
     return error("Please load the Zygote.jl package.")

test/test_zygote_gradient_wrapper.jl

@@ -25,6 +25,9 @@
     first_partial = _zygote_gradient(log, Val(2), Val(1))
     nested = _zygote_gradient(first_partial, Val(1))
     @test repr(nested) == "∂∂₁log"
+
+    # Also should work with text/plain
+    @test repr(MIME"text/plain", nested) == "∂∂₁log"
 end
 
 @testitem "ZygoteGradient evaluation" begin