custom printing

ITensor · Oct 29, 2024 · a184141 · a184141
1 parent 3dcdd20
commit a184141
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl b/NDTensors/src/lib/GradedAxes/src/gradedunitrange.jl
@@ -51,6 +51,12 @@ struct GradedOneTo{T,BlockLasts<:Vector{T}} <: AbstractGradedUnitRange{T,BlockLa
   end
 end
 
+function Base.show(io::IO, mimetype::MIME"text/plain", g::AbstractGradedUnitRange)
+  v = blocklabels(g) .=> Int.(blocklengths(g))
+  println(typeof(g))
+  return show(io, mimetype, v)
+end
+
 # == is just a range comparison that ignores labels. Need dedicated function to check equality.
 struct NoLabel end
 blocklabels(r::AbstractUnitRange) = Fill(NoLabel(), blocklength(r))