Show more digits of step size in progress_bar

ricardoV94 · ricardoV94 · commit dc7cfeeaba98 · 2025-07-24T19:45:10.000+02:00
diff --git a/pymc/step_methods/hmc/nuts.py b/pymc/step_methods/hmc/nuts.py
@@ -236,7 +236,7 @@ def competence(var, has_grad):
     def _progressbar_config(n_chains=1):
         columns = [
             TextColumn("{task.fields[divergences]}", table_column=Column("Divergences", ratio=1)),
-            TextColumn("{task.fields[step_size]:0.2f}", table_column=Column("Step size", ratio=1)),
+            TextColumn("{task.fields[step_size]:0.3f}", table_column=Column("Step size", ratio=1)),
             TextColumn("{task.fields[tree_size]}", table_column=Column("Grad evals", ratio=1)),
         ]
 

Original file line number	Diff line number	Diff line change
`@@ -236,7 +236,7 @@ def competence(var, has_grad):`
`236`	`236`	`def _progressbar_config(n_chains=1):`
`237`	`237`	`columns = [`
`238`	`238`	`TextColumn("{task.fields[divergences]}", table_column=Column("Divergences", ratio=1)),`
`239`		`- TextColumn("{task.fields[step_size]:0.2f}", table_column=Column("Step size", ratio=1)),`
	`239`	`+ TextColumn("{task.fields[step_size]:0.3f}", table_column=Column("Step size", ratio=1)),`
`240`	`240`	`TextColumn("{task.fields[tree_size]}", table_column=Column("Grad evals", ratio=1)),`
`241`	`241`	`]`
`242`	`242`