File tree Expand file tree Collapse file tree 3 files changed +6
-7
lines changed
Expand file tree Collapse file tree 3 files changed +6
-7
lines changed Original file line number Diff line number Diff line change 4848 fi
4949 pip install torch==${torch_ver} --index-url https://download.pytorch.org/whl/cpu # transformers requires torch
5050 pip install paddlepaddle==3.0.0
51- pip install vllm pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
51+ pip install pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
5252 - name : Build Package
5353 run : |
5454 pip install .
Original file line number Diff line number Diff line change 4747 torch_ver=2.6
4848 fi
4949 pip install torch==${torch_ver} --index-url https://download.pytorch.org/whl/cpu
50- pip install vllm pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
50+ pip install pytest pytest-cov setuptools_scm safetensors transformers==${tf_ver} numpy==${npy_ver}
5151 - name : Build package
5252 run : |
5353 pip install .
5858 mkdir -p /tmp/pytest-log
5959 export TEST_FASTSAFETENSORS_FRAMEWORK=pytorch
6060 COVERAGE_FILE=.coverage_0 pytest -s --cov=${LIBDIR} test_fastsafetensors.py > /tmp/pytest-log/0.log 2>&1
61- COVERAGE_FILE=.coverage_1 pytest -s --cov=${LIBDIR} test_vllm.py > /tmp/pytest-log/1.log 2>&1
62- COVERAGE_FILE=.coverage_2 torchrun --nnodes=2 --master_addr=0.0.0.0 --master_port=1234 --node_rank=0 test_multi.py --cov=${LIBDIR} -s test_multi.py > /tmp/pytest-log/2.log 2>&1 &
63- COVERAGE_FILE=.coverage_3 torchrun --nnodes=2 --master_addr=0.0.0.0 --master_port=1234 --node_rank=1 test_multi.py --cov=${LIBDIR} -s test_multi.py > /tmp/pytest-log/3.log 2>&1
61+ COVERAGE_FILE=.coverage_1 torchrun --nnodes=2 --master_addr=0.0.0.0 --master_port=1234 --node_rank=0 test_multi.py --cov=${LIBDIR} -s test_multi.py > /tmp/pytest-log/1.log 2>&1 &
62+ COVERAGE_FILE=.coverage_2 torchrun --nnodes=2 --master_addr=0.0.0.0 --master_port=1234 --node_rank=1 test_multi.py --cov=${LIBDIR} -s test_multi.py > /tmp/pytest-log/2.log 2>&1
6463 coverage combine .coverage_*
6564 coverage html
6665 mv htmlcov /tmp/pytest-log
Original file line number Diff line number Diff line change @@ -68,8 +68,8 @@ format:
6868
6969.PHONY : lint
7070lint :
71- black --check .
72- isort --check-only .
71+ black .
72+ isort .
7373 flake8 . --select=E9,F63,F7,F82
7474 mypy . --ignore-missing-imports
7575
You can’t perform that action at this time.
0 commit comments