From d1f585f80ae6845a1d15e41d61a49eadd6475f9a Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=B5=AE=E6=A2=A6?= <46097299+frozenleaves@users.noreply.github.com> Date: Thu, 27 Nov 2025 17:59:42 +0800 Subject: [PATCH] [test] update test cmd (#9544) Co-authored-by: frozenleaves Co-authored-by: Yaowei Zheng --- .github/workflows/tests.yml | 2 ++ Makefile | 2 +- src/llamafactory/v1/extras/__init__.py | 1 + 3 files changed, 4 insertions(+), 1 deletion(-) create mode 100644 src/llamafactory/v1/extras/__init__.py diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml index fe5859e5..d1f62931 100644 --- a/.github/workflows/tests.yml +++ b/.github/workflows/tests.yml @@ -8,6 +8,7 @@ on: paths: - "**/*.py" - "requirements.txt" + - "Makefile" - ".github/workflows/*.yml" pull_request: branches: @@ -15,6 +16,7 @@ on: paths: - "**/*.py" - "requirements.txt" + - "Makefile" - ".github/workflows/*.yml" jobs: diff --git a/Makefile b/Makefile index d4ca38e2..fda05fa7 100644 --- a/Makefile +++ b/Makefile @@ -21,4 +21,4 @@ style: ruff format $(check_dirs) test: - CUDA_VISIBLE_DEVICES= WANDB_DISABLED=true pytest -vv tests/ + CUDA_VISIBLE_DEVICES= ASCEND_RT_VISIBLE_DEVICES=0 WANDB_DISABLED=true pytest -vv --import-mode=importlib tests/ tests_v1/ diff --git a/src/llamafactory/v1/extras/__init__.py b/src/llamafactory/v1/extras/__init__.py new file mode 100644 index 00000000..8b137891 --- /dev/null +++ b/src/llamafactory/v1/extras/__init__.py @@ -0,0 +1 @@ +