Skip to content

Commit

Permalink
ressurect tqdm
Browse files Browse the repository at this point in the history
  • Loading branch information
xgarrido committed Sep 27, 2023
1 parent 7011ab3 commit 8a1457b
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion cobaya_utilities/plots.py
Original file line number Diff line number Diff line change
Expand Up @@ -178,6 +178,7 @@ def get_mc_samples(
set the name of the first sample to return
"""
from getdist.plots import loadMCSamples
from tqdm.auto import tqdm

selected = selected or list(mcmc_samples.keys())
excluded = excluded or []
Expand All @@ -195,7 +196,8 @@ def get_mc_samples(
selected = [select_first] + selected

samples, labels, colors = [], [], []
for name in selected:
for name in (pbar := tqdm(selected)):
pbar.set_description(f"Loading '{name}'")
value = mcmc_samples[name]
path = _get_path(name, value)
if isinstance(value, dict):
Expand Down

0 comments on commit 8a1457b

Please sign in to comment.