diff --git a/.github/workflows/pr_tests.yml b/.github/workflows/pr_tests.yml index 66ce53d576..8df8dd0146 100644 --- a/.github/workflows/pr_tests.yml +++ b/.github/workflows/pr_tests.yml @@ -136,6 +136,8 @@ jobs: - name: Run fast PyTorch Pipeline CPU tests if: ${{ matrix.config.framework == 'pytorch_pipelines' }} + env: + PYTEST_TIMEOUT: 300 run: | pytest -n 8 --max-worker-restart=0 --dist=loadfile \ -k "not Onnx" \ @@ -152,6 +154,8 @@ jobs: - name: Run example PyTorch CPU tests if: ${{ matrix.config.framework == 'pytorch_examples' }} + env: + PYTEST_TIMEOUT: 300 run: | uv pip install ".[training]" pytest -n 4 --max-worker-restart=0 --dist=loadfile \ diff --git a/tests/others/test_utils.py b/tests/others/test_utils.py index d1a59cec52..3741b1ee70 100755 --- a/tests/others/test_utils.py +++ b/tests/others/test_utils.py @@ -175,7 +175,7 @@ class TestDeprecate: with pytest.warns(FutureWarning) as warning: deprecate(("deprecated_arg", self.higher_version, "This message is better!!!"), standard_warn=False) assert str(warning[0].message) == "This message is better!!!" - assert "diffusers/tests/others/test_utils.py" in warning[0].filename + assert "tests/others/test_utils.py" in warning[0].filename def test_deprecate_testing_utils_module(self): import diffusers.utils.testing_utils