Rename make_update_stats_functions to _make_progressbar_update_functions

Author: ricardoV94

pymc/progress_bar.py

@@ -209,8 +209,8 @@ def __init__(
 
         When sampling, Step classes are responsible for computing and exposing statistics that can be reported on
         progress bars. Each Step implements two class methods: :meth:`pymc.step_methods.BlockedStep._progressbar_config`
-        and :meth:`pymc.step_methods.BlockedStep._make_update_stats_function`. `_progressbar_config` reports which
-        columns should be displayed on the progress bar, and `_make_update_stats_function` computes the statistics
+        and :meth:`pymc.step_methods.BlockedStep._make_progressbar_update_functions`. `_progressbar_config` reports which
+        columns should be displayed on the progress bar, and `_make_progressbar_update_functions` computes the statistics
         that will be displayed on the progress bar.
 
         Parameters
@@ -281,7 +281,7 @@ def __init__(
             progressbar_theme=progressbar_theme,
         )
         self.progress_stats = progress_stats
-        self.update_stats_functions = step_method._make_update_stats_functions()
+        self.update_stats_functions = step_method._make_progressbar_update_functions()
 
         self._show_progress = show_progress
         self.divergences = 0

pymc/step_methods/compound.py

@@ -189,7 +189,7 @@ def _progressbar_config(n_chains=1):
         return columns, stats
 
     @staticmethod
-    def _make_update_stats_functions():
+    def _make_progressbar_update_functions():
         def update_stats(step_stats):
             return step_stats
 
@@ -332,10 +332,10 @@ def _progressbar_config(self, n_chains=1):
 
         return columns, stats
 
-    def _make_update_stats_functions(self):
+    def _make_progressbar_update_functions(self):
         update_functions = []
         for method in self.methods:
-            update_functions.extend(method._make_update_stats_functions())
+            update_functions.extend(method._make_progressbar_update_functions())
         return update_functions
 
 

pymc/step_methods/hmc/nuts.py

@@ -248,7 +248,7 @@ def _progressbar_config(n_chains=1):
         return columns, stats
 
     @staticmethod
-    def _make_update_stats_functions():
+    def _make_progressbar_update_functions():
         def update_stats(stats):
             return {key: stats[key] for key in ("diverging", "step_size", "tree_size")}
 

pymc/step_methods/metropolis.py

@@ -346,7 +346,7 @@ def _progressbar_config(n_chains=1):
         return columns, stats
 
     @staticmethod
-    def _make_update_stats_functions():
+    def _make_progressbar_update_functions():
         def update_stats(step_stats):
             return {
                 "accept_rate" if key == "accept" else key: step_stats[key]

pymc/step_methods/slicer.py

@@ -212,7 +212,7 @@ def _progressbar_config(n_chains=1):
         return columns, stats
 
     @staticmethod
-    def _make_update_stats_functions():
+    def _make_progressbar_update_functions():
         def update_stats(step_stats):
             return {key: step_stats[key] for key in {"tune", "nstep_out", "nstep_in"}}