diff --git a/tests/tests_fabric/utilities/test_seed.py b/tests/tests_fabric/utilities/test_seed.py index 0973709bf84fd..4a948a5f98736 100644 --- a/tests/tests_fabric/utilities/test_seed.py +++ b/tests/tests_fabric/utilities/test_seed.py @@ -1,5 +1,6 @@ import os import random +import warnings from unittest import mock from unittest.mock import Mock @@ -30,9 +31,9 @@ def test_seed_stays_same_with_multiple_seed_everything_calls(): seed_everything() initial_seed = os.environ.get("PL_GLOBAL_SEED") - with pytest.warns(None) as record: + with warnings.catch_warnings(): + warnings.simplefilter("error") seed_everything() - assert not record # does not warn seed = os.environ.get("PL_GLOBAL_SEED") assert initial_seed == seed