Merge pull request #520 from CUQI-DTU/add_prefix_to_sampling_progress…

…_516 Add prefix to sampling progress
CUQI-DTU · Sep 12, 2024 · f2fb340 · f2fb340
2 parents 464aa14 + d8364e3
commit f2fb340
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/cuqi/experimental/mcmc/_gibbs.py b/cuqi/experimental/mcmc/_gibbs.py
@@ -161,7 +161,7 @@ def sample(self, Ns) -> 'HybridGibbs':
 
         """
 
-        for _ in tqdm(range(Ns)):
+        for _ in tqdm(range(Ns), "Sample: "):
 
             self.step()
 
@@ -184,7 +184,7 @@ def warmup(self, Nb, tune_freq=0.1) -> 'HybridGibbs':
 
         tune_interval = max(int(tune_freq * Nb), 1)
 
-        for idx in tqdm(range(Nb)):
+        for idx in tqdm(range(Nb), "Warmup: "):
 
             self.step()
 

diff --git a/cuqi/experimental/mcmc/_sampler.py b/cuqi/experimental/mcmc/_sampler.py
@@ -220,7 +220,7 @@ def sample(self, Ns, batch_size=0, sample_path='./CUQI_samples/') -> 'Sampler':
         if hasattr(self, "_pre_sample"): self._pre_sample()
 
         # Draw samples
-        for _ in tqdm( range(Ns) ):
+        for _ in tqdm( range(Ns), "Sample: "):
 
             # Perform one step of the sampler
             acc = self.step()
@@ -260,7 +260,7 @@ def warmup(self, Nb, tune_freq=0.1) -> 'Sampler':
         if hasattr(self, "_pre_warmup"): self._pre_warmup()
 
         # Draw warmup samples with tuning
-        for idx in tqdm(range(Nb)):
+        for idx in tqdm(range(Nb), "Warmup: "):
 
             # Perform one step of the sampler
             acc = self.step()