Skip to content

Commit 1924f9c

Browse files
authored
Adapt sample to use new progress bar (#259)
1 parent ddd70a1 commit 1924f9c

File tree

1 file changed

+6
-2
lines changed

1 file changed

+6
-2
lines changed

pymc_bart/pgbart.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -429,9 +429,13 @@ def competence(var: pm.Distribution, has_grad: bool) -> Competence:
429429
return Competence.INCOMPATIBLE
430430

431431
@staticmethod
432-
def _make_update_stats_functions():
432+
def _progressbar_config(n_chains=1):
433+
return [], {}
434+
435+
@staticmethod
436+
def _make_progressbar_update_functions():
433437
def update_stats(step_stats):
434-
return {key: step_stats[key] for key in ("variable_inclusion", "tune")}
438+
return {"tune": step_stats.get("tune", False)}
435439

436440
return (update_stats,)
437441

0 commit comments

Comments
 (0)