Use uv pip compile in Update Requirements GitHub Action (#3452)

Author: yifanmai

.github/workflows/update-requirements.yml

@@ -32,23 +32,8 @@ jobs:
           cache: pip
       - name: Install uv
         run: pip install uv
-      - name: Install PyTorch manually
-        # PyTorch has to be installed manually in a separate step with --no-cache-dir
-        # to avoid pip getting killed because PyTorch is too big
-        # See: https://stackoverflow.com/a/54329850
-        run: uv pip install --system torch==2.2.2 torchvision==0.17.2 --index-url https://download.pytorch.org/whl/cpu --no-cache-dir
-      - name: Upgrade dependencies
-        run: uv pip install --system --upgrade -e .[all,dev] torch==2.2.2 torchvision==0.17.2
-      - name: Verify dependencies
-        run: uv pip check && pip check
-      - name: Write header to requirements.txt
-        run: echo '# This file is automatically generated by GitHub Actions and contains pinned versions for all transitive Python dependencies. Do not modify this file!' > requirements.txt
-      - name: Write dependencies to requirements.txt
-        # grep is used to exclude dependencies installed in editable mode
-        # and dependencies installed from outside PyPI.
-        # sed is used to loosen the version matches for PyTorch
-        # so that any CUDA version matches.
-        run: pip freeze | grep -v -E "( @ |^-e)" | sed -E 's/==([A-Za-z0-9.]+)\+[A-Za-z0-9.]+/~=\1/' >> requirements.txt
+      - name: Resolve dependencies
+        run: uv pip compile setup.cfg --extra all --extra dev --universal --no-annotate --constraints constraints.txt -o requirements.txt
         # Need to manually run tests here because the pull request opened later will not
         # run the test workflow.
         #
@@ -59,6 +44,10 @@ jobs:
         #     if a workflow run pushes code using the repository's GITHUB_TOKEN, a new workflow
         #     will not run even when the repository contains a workflow configured to run when
         #     push events occur.
+      - name: Install HELM 
+        run: ./install-dev.sh
+      - name: Verify dependencies
+        run: pip check
       - name: Run linter
         run: ./pre-commit.sh
       - name: Run tests