Skip to content

Commit 959f073

Browse files
Add progressbar defaults to BlockedStep ABC
1 parent 1c5b734 commit 959f073

File tree

1 file changed

+14
-0
lines changed

1 file changed

+14
-0
lines changed

pymc/step_methods/compound.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -181,6 +181,20 @@ def __new__(cls, *args, **kwargs):
181181
step.__newargs = (vars, *args), kwargs
182182
return step
183183

184+
@staticmethod
185+
def _progressbar_config(n_chains=1):
186+
columns = []
187+
stats = {}
188+
189+
return columns, stats
190+
191+
@staticmethod
192+
def _make_update_stats_function():
193+
def update_stats(stats, step_stats, chain_idx):
194+
return stats
195+
196+
return update_stats
197+
184198
# Hack for creating the class correctly when unpickling.
185199
def __getnewargs_ex__(self):
186200
return self.__newargs

0 commit comments

Comments
 (0)