From 0714a49227bbd53936211daee8f60d8eb7293ffd Mon Sep 17 00:00:00 2001 From: Wing Lian Date: Mon, 13 Jan 2025 21:38:26 -0500 Subject: [PATCH] move relora test so it runs in a single test thread --- cicd/cicd.sh | 3 ++- tests/e2e/solo/__init__.py | 0 tests/e2e/{patched => solo}/test_relora_llama.py | 0 3 files changed, 2 insertions(+), 1 deletion(-) create mode 100644 tests/e2e/solo/__init__.py rename tests/e2e/{patched => solo}/test_relora_llama.py (100%) diff --git a/cicd/cicd.sh b/cicd/cicd.sh index 91926127f..34a30db44 100755 --- a/cicd/cicd.sh +++ b/cicd/cicd.sh @@ -6,5 +6,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 /workspace/axolotl/tests/e2e/patched/ +pytest -v --durations=10 -n1 /workspace/axolotl/tests/e2e/solo/ pytest -v --durations=10 /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/ +pytest -v --durations=10 --ignore=tests/e2e/solo/ --ignore=tests/e2e/patched/ --ignore=tests/e2e/multigpu/ --ignore=tests/e2e/integrations/ /workspace/axolotl/tests/e2e/ diff --git a/tests/e2e/solo/__init__.py b/tests/e2e/solo/__init__.py new file mode 100644 index 000000000..e69de29bb diff --git a/tests/e2e/patched/test_relora_llama.py b/tests/e2e/solo/test_relora_llama.py similarity index 100% rename from tests/e2e/patched/test_relora_llama.py rename to tests/e2e/solo/test_relora_llama.py