From f2bd07044e330cd77970f2adebc224ff62cec649 Mon Sep 17 00:00:00 2001 From: Debanjum Date: Fri, 1 Aug 2025 23:23:44 -0700 Subject: [PATCH] Speed up github workflows by not installing cuda server dependencies - CI runners don't have GPUs - Pytorch related Nvidia cuda packages are not required for testing, evals or pre-commit checks. - Avoiding these massive downloads should speed up workflow run. --- .github/workflows/pre-commit.yml | 4 ++++ .github/workflows/run_evals.yml | 4 ++++ .github/workflows/test.yml | 3 ++- 3 files changed, 10 insertions(+), 1 deletion(-) diff --git a/.github/workflows/pre-commit.yml b/.github/workflows/pre-commit.yml index 5b529b98..6352e688 100644 --- a/.github/workflows/pre-commit.yml +++ b/.github/workflows/pre-commit.yml @@ -48,6 +48,10 @@ jobs: sudo apt update && sudo apt install -y libegl1 - name: ⬇️ Install Application + env: + UV_INDEX: "https://download.pytorch.org/whl/cpu" + UV_INDEX_STRATEGY: "unsafe-best-match" + CUDA_VISIBLE_DEVICES: "" run: uv sync --all-extras - name: 🌡️ Validate Application diff --git a/.github/workflows/run_evals.yml b/.github/workflows/run_evals.yml index 62bc4af6..ccab4034 100644 --- a/.github/workflows/run_evals.yml +++ b/.github/workflows/run_evals.yml @@ -135,6 +135,10 @@ jobs: git clone https://github.com/khoj-ai/terrarium.git && cd terrarium && npm install --legacy-peer-deps && mkdir pyodide_cache - name: ⬇️ Install Application + env: + UV_INDEX: "https://download.pytorch.org/whl/cpu" + UV_INDEX_STRATEGY: "unsafe-best-match" + CUDA_VISIBLE_DEVICES: "" run: | sed -i 's/dynamic = \["version"\]/version = "${{ steps.hatch.outputs.version }}"/' pyproject.toml uv sync --all-extras diff --git a/.github/workflows/test.yml b/.github/workflows/test.yml index d1d85dec..c4c5628b 100644 --- a/.github/workflows/test.yml +++ b/.github/workflows/test.yml @@ -76,7 +76,8 @@ jobs: - name: ⬇️ Install Application env: - PIP_EXTRA_INDEX_URL: "https://download.pytorch.org/whl/cpu https://abetlen.github.io/llama-cpp-python/whl/cpu" + UV_INDEX: "https://download.pytorch.org/whl/cpu" + UV_INDEX_STRATEGY: "unsafe-best-match" CUDA_VISIBLE_DEVICES: "" run: sed -i 's/dynamic = \["version"\]/version = "0.0.0"/' pyproject.toml && uv sync --all-extras