Skip to content

Commit c84d3b5

Browse files
committed
Improve printer code style
1 parent 1149324 commit c84d3b5

File tree

1 file changed

+10
-4
lines changed

1 file changed

+10
-4
lines changed

src/solvers/defaults.jl

Lines changed: 10 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
using Printf: @printf
1+
using Printf: @printf, @sprintf
22
using ITensorMPS: maxlinkdim
33
default_outputlevel() = 0
44
default_nsites() = 2
@@ -7,6 +7,10 @@ default_extracter() = default_extracter
77
default_inserter() = default_inserter
88
default_checkdone() = (; kws...) -> false
99
default_transform_operator() = nothing
10+
11+
format(x) = @sprintf("%s", x)
12+
format(x::AbstractFloat) = @sprintf("%.1E", x)
13+
1014
function default_region_printer(;
1115
inserter_kwargs,
1216
outputlevel,
@@ -21,9 +25,11 @@ function default_region_printer(;
2125
region = first(sweep_plan[which_region_update])
2226
@printf("Sweep %d, region=%s \n", which_sweep, region)
2327
print(" Truncated using")
24-
haskey(inserter_kwargs, :cutoff) && @printf(" cutoff=%.1E", inserter_kwargs.cutoff)
25-
haskey(inserter_kwargs, :maxdim) && @printf(" maxdim=%d", inserter_kwargs.maxdim)
26-
haskey(inserter_kwargs, :mindim) && @printf(" mindim=%d", inserter_kwargs.mindim)
28+
for key in [:cutoff, :maxdim, :mindim]
29+
if haskey(inserter_kwargs, key)
30+
print(" ", key, "=", format(inserter_kwargs[key]))
31+
end
32+
end
2733
println()
2834
if spec != nothing
2935
@printf(

0 commit comments

Comments
 (0)