Fix public github CI for axlearn.

GitOrigin-RevId: de8a20fc42fa52408ba10897e7bb074a028e32f2

Author: raytomato

.github/workflows/pre-commit.yml

@@ -13,11 +13,16 @@ jobs:
       - uses: actions/checkout@v4
       - uses: actions/setup-python@v5
         with:
-          python-version: '3.10'
+          python-version: '3.12'
           cache: 'pip'
       - run: pip install --upgrade pip
+      - run: pip install uv
+      - run: uv venv
+      - run: |
+          echo ".venv/bin" >> $GITHUB_PATH
+          echo "VIRTUAL_ENV=.venv" >> $GITHUB_ENV
       # TODO(markblee): Remove gcp,vertexai_tensorboard from CI. (needed by pytype)
-      - run: pip install '.[core,audio,orbax,dev,gcp,vertexai_tensorboard,open_api]'
+      - run: uv pip install '.[core,audio,orbax,dev,gcp,vertexai_tensorboard,open_api]'
       # pylint uses approx 12GB of memory during this run, look into split to decrease?
       - run: |
           # Start memory monitor as a background process.