From 2852eb3004249540fdc4a5f206c34a58d9f08855 Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Thu, 12 Dec 2024 22:12:39 -0500 Subject: [PATCH] run the patched e2e tests in pytest-forked --- cicd/cicd.sh | 4 ++-- requirements-tests.txt | 1 + 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cicd/cicd.sh b/cicd/cicd.sh index 88defdd0e4..3ad12e4694 100755 --- a/cicd/cicd.sh +++ b/cicd/cicd.sh @@ -5,6 +5,6 @@ python -c "import torch; assert '$PYTORCH_VERSION' in torch.__version__" pytest -v --durations=10 -n8 --ignore=tests/e2e/ --ignore=tests/patched/ /workspace/axolotl/tests/ # pytest -v --durations=10 -n8 --dist loadfile /workspace/axolotl/tests/patched/ -pytest -v --durations=10 -n1 --dist loadfile /workspace/axolotl/tests/e2e/patched/ -pytest -v --durations=10 -n1 --dist loadfile /workspace/axolotl/tests/e2e/integrations/ +pytest --forked -v --durations=10 -n1 --dist loadfile /workspace/axolotl/tests/e2e/patched/ +pytest --forked -v --durations=10 -n1 --dist loadfile /workspace/axolotl/tests/e2e/integrations/ pytest -v --durations=10 --ignore=tests/e2e/patched/ --ignore=tests/e2e/multigpu/ --ignore=tests/e2e/integrations/ /workspace/axolotl/tests/e2e/ diff --git a/requirements-tests.txt b/requirements-tests.txt index a13f739231..6b7a484f34 100644 --- a/requirements-tests.txt +++ b/requirements-tests.txt @@ -1,5 +1,6 @@ pytest pytest-xdist +pytest-forked pytest-retry pytest-sugar tbparse