Skip to content

Commit c18e507

Browse files
authored
[Python] Update system platform requirement for flashinfer (#3321)
This PR updates the dependency system requirement for flashinfer-python so it won't be treated as a dependency on Mac.
1 parent 24ab415 commit c18e507

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

ci/task/pylint.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,10 +9,10 @@ export PYTHONPATH="./python":${PYTHONPATH:-""}
99
if [[ -n ${MLC_CI_SETUP_DEPS:-} ]]; then
1010
echo "MLC_CI_SETUP_DEPS=1 start setup deps"
1111
# TVM Unity is a dependency to this testing
12-
pip install --quiet --pre -U --no-index -f https://mlc.ai/wheels mlc-ai-nightly-cpu
12+
pip install --pre -U -f https://mlc.ai/wheels mlc-ai-nightly-cpu
1313
pip install apache-tvm-ffi
1414
pip install requests triton
15-
pip install --quiet --pre -U cuda-python
15+
pip install --pre -U cuda-python
1616
fi
1717

1818
pylint --jobs $NUM_THREADS ./python/

pyproject.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ dependencies = [
3838
"apache-tvm-ffi",
3939
"datasets",
4040
"fastapi",
41-
"flashinfer-python==0.2.14",
41+
"flashinfer-python==0.2.14; sys_platform == 'linux'",
4242
"ml_dtypes>=0.5.1",
4343
"openai",
4444
"pandas",

0 commit comments

Comments
 (0)