Refresh packet progress bar after every iteration

tardis-sn · Mar 9, 2023 · 46a0f2e · 46a0f2e
1 parent 239e04b
commit 46a0f2e
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/tardis/montecarlo/montecarlo_numba/base.py b/tardis/montecarlo/montecarlo_numba/base.py
@@ -29,7 +29,7 @@
 )
 from tardis.montecarlo.montecarlo_numba import njit_dict
 from numba.typed import List
-from tardis.util.base import update_iterations_pbar, update_packet_pbar
+from tardis.util.base import update_iterations_pbar, update_packet_pbar, refresh_packet_pbar
 
 
 def montecarlo_radial1d(
@@ -128,6 +128,7 @@ def montecarlo_radial1d(
             virt_packet_last_line_interaction_out_id
         ).ravel()
     update_iterations_pbar(1)
+    refresh_packet_pbar()
     # Condition for Checking if RPacket Tracking is enabled
     if montecarlo_configuration.RPACKET_TRACKING:
         runner.rpacket_tracker = rpacket_trackers

diff --git a/tardis/util/base.py b/tardis/util/base.py
@@ -699,6 +699,14 @@ def update_packet_pbar(i, current_iteration, no_of_packets, total_iterations):
     packet_pbar.update(i)
 
 
+def refresh_packet_pbar():
+    """
+    Refresh packet progress bar after each iteration.
+
+    """
+    packet_pbar.refresh()
+
+
 def update_iterations_pbar(i):
     """
     Update progress bar for each iteration.