Print description in AbstractTrees.printnode()

SciML · Oct 18, 2024 · 77025da · 77025da
1 parent 699fa62
commit 77025da
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/systems/abstractsystem.jl b/src/systems/abstractsystem.jl
@@ -2901,11 +2901,12 @@ function Base.showerror(io::IO, e::HybridSystemNotSupportedException)
     print(io, "HybridSystemNotSupportedException: ", e.msg)
 end
 
-function AbstractTrees.children(sys::ModelingToolkit.AbstractSystem)
+function AbstractTrees.children(sys::AbstractSystem)
     ModelingToolkit.get_systems(sys)
 end
-function AbstractTrees.printnode(io::IO, sys::ModelingToolkit.AbstractSystem)
+function AbstractTrees.printnode(io::IO, sys::AbstractSystem; describe = true)
     print(io, nameof(sys))
+    describe && !isempty(get_description(sys)) && print(io, ": ", get_description(sys))
 end
 function Base.IteratorEltype(::Type{<:TreeIterator{ModelingToolkit.AbstractSystem}})
     Base.HasEltype()