diff --git a/src/printing.jl b/src/printing.jl index 2d1715b..5935f9f 100644 --- a/src/printing.jl +++ b/src/printing.jl @@ -63,3 +63,11 @@ function Base.show(io::IO, ::MIME"text/plain", g::Chain) print(io, ")") end +function Base.show(io::IO, x::IRCode) + if x.stmts.stmt[1] isa Core.ReturnNode + printstyled(io, nothing) + else + printstyled(io, x) + end +end +