diff --git a/.github/workflows/ci-examples-app.yml b/.github/workflows/ci-examples-app.yml index bd2b002dac31e..dd2239bcc31a6 100644 --- a/.github/workflows/ci-examples-app.yml +++ b/.github/workflows/ci-examples-app.yml @@ -118,7 +118,7 @@ jobs: run: | python -m coverage run --source ${{ env.COVERAGE_SCOPE }} \ -m pytest -m "not cloud" integrations_app \ - --timeout=300 --durations=0 -vvvv + --timeout=120 --durations=0 -vvvv - name: Statistics if: success() diff --git a/.github/workflows/ci-tests-app.yml b/.github/workflows/ci-tests-app.yml index b9045e1ed5740..1afe96fbead60 100644 --- a/.github/workflows/ci-tests-app.yml +++ b/.github/workflows/ci-tests-app.yml @@ -124,7 +124,7 @@ jobs: AWS_DEFAULT_REGION: us-east-1 run: | python -m coverage run --source ${{ env.COVERAGE_SCOPE }} \ - -m pytest -m "not cloud" tests_app --timeout=300 -vvvv --durations=50 + -m pytest -m "not cloud" tests_app --timeout=120 -vvvv --durations=50 - name: Statistics if: success() diff --git a/tests/tests_app/utilities/test_proxies.py b/tests/tests_app/utilities/test_proxies.py index 8dbfe1b06b871..d62cf2d6a2344 100644 --- a/tests/tests_app/utilities/test_proxies.py +++ b/tests/tests_app/utilities/test_proxies.py @@ -76,6 +76,7 @@ def proxy_setattr(): ], ) @mock.patch("lightning.app.utilities.proxies._Copier", MagicMock()) +@pytest.mark.flaky(reruns=3) @pytest.mark.xfail(sys.platform == "win32", strict=False, reason="Fix this on Windows") # TODO @ethanwharris def test_work_runner(parallel, cache_calls, *_): """This test validates the `WorkRunner` runs the work.run method and properly populates the `delta_queue`, diff --git a/tests/tests_pytorch/utilities/test_all_gather_grad.py b/tests/tests_pytorch/utilities/test_all_gather_grad.py index 2e976bb82e7e1..e288a591ba505 100644 --- a/tests/tests_pytorch/utilities/test_all_gather_grad.py +++ b/tests/tests_pytorch/utilities/test_all_gather_grad.py @@ -13,6 +13,7 @@ # limitations under the License. import numpy as np +import pytest import torch from lightning.pytorch import Trainer @@ -45,6 +46,7 @@ def all_gather_ddp_spawn_fn(strategy): @RunIf(skip_windows=True) +@pytest.mark.flaky(reruns=3) def test_all_gather_ddp_spawn(): spawn_launch(all_gather_ddp_spawn_fn, [torch.device("cpu")] * 3)