Merge pull request festim-dev#759 from RemDelaporteMathurin/tqdm-auto

`tqdm.auto` instead of `tqdm.autonotebook` + refresh
KulaginVladimir · Apr 22, 2024 · c7a4641 · c7a4641
2 parents b85ad9e + fa260c3
commit c7a4641
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/festim/hydrogen_transport_problem.py b/festim/hydrogen_transport_problem.py
@@ -6,7 +6,7 @@
 import ufl
 from mpi4py import MPI
 import numpy as np
-import tqdm.autonotebook
+import tqdm.auto as tqdm
 import festim as F
 
 from dolfinx.mesh import meshtags
@@ -753,13 +753,14 @@ def run(self):
 
         if self.settings.transient:
             # Solve transient
-            self.progress = tqdm.autonotebook.tqdm(
+            self.progress = tqdm.tqdm(
                 desc="Solving H transport problem",
                 total=self.settings.final_time,
                 unit_scale=True,
             )
             while self.t.value < self.settings.final_time:
                 self.iterate()
+            self.progress.refresh()  # refresh progress bar to show 100%
         else:
             # Solve steady-state
             self.solver.solve(self.u)