From 73aa00ec0a960f683b80f150fdc7c2a2ee147dea Mon Sep 17 00:00:00 2001
From: ur4t <46435411+ur4t@users.noreply.github.com>
Date: Tue, 17 Dec 2024 12:05:13 +0800
Subject: [PATCH] bugfix: Fix setuptools-scm related issues. (#668)

---
 docs/requirements.txt         | 3 ++-
 scripts/run-ci-build-wheel.sh | 4 ++--
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/docs/requirements.txt b/docs/requirements.txt
index 479aea944..c8ce7afe1 100644
--- 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
index 577de984a..0f19d65b1 100644
--- 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