Fix show for tangent types with free diff level

Follows up #136
JuliaDiff · Apr 9, 2023 · 44c0094 · 44c0094
1 parent 3751c0f
commit 44c0094
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/src/tangent.jl b/src/tangent.jl
@@ -261,19 +261,20 @@ wrapper_name(::Type{<:AbstractZeroBundle}) = "AbstractZeroBundle"
 
 function Base.show(io::IO, T::Type{<:AbstractZeroBundle{N, B}}) where {N,B}
     print(io, wrapper_name(T))
-    print(io, "{$N, ")
+    print(io, @isdefined(N) ? "{$N, " : "{N, ")
     show(io, B)
     print(io, "}")
 end
 
 function Base.show(io::IO, T::Type{<:AbstractZeroBundle{N}}) where {N}
     print(io, wrapper_name(T))
-    print(io, "{$N}")
+    @isdefined(N) && print(io, "{$N}")
 end
 
 function Base.show(io::IO, t::AbstractZeroBundle{N}) where N
     print(io, wrapper_name(typeof(t)))
-    print(io, "{$N}(")
+    @isdefined(N) && print(io, "{$N}")
+    print(io, "(")
     show(io, t.primal)
     print(io, ")")
 end

diff --git a/test/tangent.jl b/test/tangent.jl
@@ -18,6 +18,8 @@ using Test
         @test repr(ZeroBundle{1}) == "ZeroBundle{1}"
         @test repr(ZeroBundle{1, Float64}) == "ZeroBundle{1, Float64}"
 
+        @test repr((ZeroBundle{N, Float64} where N).body) == "ZeroBundle{N, Float64}"
+
         @test repr(typeof(DNEBundle{1}(getfield))) == "DNEBundle{1, typeof(getfield)}"
     end
 end