diff --git a/tests/callbacks/test_rich_model_summary.py b/tests/callbacks/test_rich_model_summary.py index 00d39b8b71a150..c596557eed0dcf 100644 --- a/tests/callbacks/test_rich_model_summary.py +++ b/tests/callbacks/test_rich_model_summary.py @@ -37,7 +37,7 @@ def test_rich_progress_bar_import_error(monkeypatch): monkeypatch.setattr(imports, "_RICH_AVAILABLE", False) with pytest.raises(ModuleNotFoundError, match="`RichModelSummary` requires `rich` to be installed."): - RichModelSummary() + RichModelSummary() @RunIf(rich=True) diff --git a/tests/callbacks/test_rich_progress_bar.py b/tests/callbacks/test_rich_progress_bar.py index 7c6d2b656d08e1..6b47e9558f92f6 100644 --- a/tests/callbacks/test_rich_progress_bar.py +++ b/tests/callbacks/test_rich_progress_bar.py @@ -89,6 +89,7 @@ def test_rich_progress_bar_import_error(monkeypatch): with pytest.raises(ModuleNotFoundError, match="`RichProgressBar` requires `rich` >= 10.2.2."): RichProgressBar() + @RunIf(rich=True) def test_rich_progress_bar_custom_theme(tmpdir): """Test to ensure that custom theme styles are used."""