From 959f0dcee2ab02aec3013aa68d9df9d78a428108 Mon Sep 17 00:00:00 2001 From: Chris Fonnesbeck Date: Sat, 12 Apr 2025 13:28:58 -0500 Subject: [PATCH] Progress bar fix for PF --- pymc_extras/inference/pathfinder/pathfinder.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/pymc_extras/inference/pathfinder/pathfinder.py b/pymc_extras/inference/pathfinder/pathfinder.py index ed30d22c5..8a73549f9 100644 --- a/pymc_extras/inference/pathfinder/pathfinder.py +++ b/pymc_extras/inference/pathfinder/pathfinder.py @@ -587,8 +587,6 @@ def bfgs_sample_sparse( # (L, N, N), (L, N, M) -> (L, N, M) + sqrt_alpha_diag @ ( - # (L, N, 2J), (L, 2J, M) -> (L, N, M) - # intermediate calcs below # (L, N, 2J), (L, 2J, 2J) -> (L, N, 2J) (Q @ (Lchol - IdN)) # (L, 2J, N), (L, N, M) -> (L, 2J, M) @@ -1565,8 +1563,9 @@ def multipath_pathfinder( task, description=desc.format(path_idx=path_idx), completed=path_idx, - refresh=True, ) + # Ensure the progress bar visually reaches 100% and shows 'Completed' + progress.update(task, completed=num_paths, description="Completed") except (KeyboardInterrupt, StopIteration) as e: # if exception is raised here, MultiPathfinderResult will collect all the successful results and report the results. User is free to abort the process earlier and the results will still be collected and return az.InferenceData. if isinstance(e, StopIteration):