bugfix: Fix setuptools-scm related issues. (#668)

flashinfer-ai · Dec 17, 2024 · 73aa00e · 73aa00e
1 parent 08405a5
commit 73aa00e
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/docs/requirements.txt b/docs/requirements.txt
@@ -1,6 +1,7 @@
 furo == 2024.8.6
+setuptools == 75.6.0
+setuptools-scm == 8.1.0
 sphinx == 8.1.3
 sphinx-reredirects == 0.1.5
 sphinx-tabs == 3.4.5
 sphinx-toolbox == 3.8.1
-setuptools-scm == 8.1.0
diff --git a/scripts/run-ci-build-wheel.sh b/scripts/run-ci-build-wheel.sh
@@ -14,8 +14,8 @@ assert_env FLASHINFER_CI_TORCH_VERSION
 assert_env FLASHINFER_CI_CUDA_VERSION
 assert_env TORCH_CUDA_ARCH_LIST
 PROJECT_ROOT="$(cd "$(dirname "$0")/.." && pwd)"
-export CONDA_pkgs_dirs=/ci-cache/conda-pkgs
-export XDG_CACHE_HOME=/ci-cache/xdg-cache
+export CONDA_pkgs_dirs="${FLASHINFER_CI_CACHE:=/ci-cache}/conda-pkgs"
+export XDG_CACHE_HOME="${FLASHINFER_CI_CACHE:=/ci-cache}/xdg-cache"
 mkdir -p "$CONDA_pkgs_dirs" "$XDG_CACHE_HOME"
 export HOME=/tmp/home
 mkdir -p $HOME