temp fix unittests

Author: GdoongMathew

src/lightning/pytorch/callbacks/progress/rich_progress.py

@@ -103,6 +103,14 @@ def start(self) -> None:
                 self.live.auto_refresh = True
                 self.live._refresh_thread.start()
 
+        def stop(self) -> None:
+            refresh_thread = self.live._refresh_thread
+            self.live.auto_refresh = refresh_thread is not None
+            super().stop()
+            if refresh_thread:
+                refresh_thread.stop()
+                refresh_thread.join()
+
         def refresh(self) -> None:
             if self.live.auto_refresh:
                 self.live._refresh_thread.refresh_cond = True

tests/tests_pytorch/callbacks/progress/test_rich_progress_bar.py

@@ -131,6 +131,8 @@ def test_rich_progress_bar_custom_theme():
         _, kwargs = mocks["ProcessingSpeedColumn"].call_args
         assert kwargs["style"] == theme.processing_speed
 
+        progress_bar.progress.live._refresh_thread.stop()
+
 
 @RunIf(rich=True)
 def test_rich_progress_bar_keyboard_interrupt(tmp_path):
@@ -176,6 +178,8 @@ def configure_columns(self, trainer):
     assert progress_bar.progress.columns[0] == custom_column
     assert len(progress_bar.progress.columns) == 2
 
+    progress_bar.progress.stop()
+
 
 @RunIf(rich=True)
 @pytest.mark.parametrize(("leave", "reset_call_count"), ([(True, 0), (False, 3)]))