Fixed Progress bar

probabl-ai · Jan 29, 2025 · 0bbb11d · 0bbb11d
1 parent 66ca391
commit 0bbb11d
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/skore/src/skore/utils/_progress_bar.py → skore/src/skore/utils/Fixed Progress bar b/skore/src/skore/utils/_progress_bar.py → skore/src/skore/utils/Fixed Progress bar
@@ -1,5 +1,5 @@
+import sys
 from functools import wraps
-
 from rich.progress import (
     BarColumn,
     Progress,
@@ -43,6 +43,7 @@ def wrapper(*args, **kwargs):
                     ),
                     TextColumn("[orange1]{task.percentage:>3.0f}%"),
                     expand=False,
+                    transient="pytest" not in sys.modules,  # Disable transient during tests
                 )
                 progress.start()
 
@@ -68,7 +69,7 @@ def wrapper(*args, **kwargs):
                             task, completed=progress.tasks[task].total, refresh=True
                         )
                     progress.stop()
-                # clean up to make object pickable
+                # Clean up to make object pickable
                 self_obj._parent_progress = None
                 self_obj._progress_info = None