Update docs workflows to use uv (#1032)

Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>

.github/workflows/codspeed.yml

@@ -13,20 +13,20 @@ concurrency:
 
 jobs:
   benchmarks:
-    runs-on: ubuntu-latest
+    runs-on: ubuntu-22.04
     steps:
       - uses: actions/checkout@v4
 
       - name: Setup Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
           python-version: "3.12"
           # Looks like it's not working very well for other people:
           # https://github.com/actions/setup-python/issues/436
           # cache: "pip"
           # cache-dependency-path: pyproject.toml
 
-      - uses: actions/cache@v3
+      - uses: actions/cache@v4
         id: cache
         with:
           path: ${{ env.pythonLocation }}
@@ -37,7 +37,7 @@ jobs:
         run: ./scripts/install_dependencies.sh
 
       - name: Run benchmarks
-        uses: CodSpeedHQ/action@v2
+        uses: CodSpeedHQ/action@v3
         with:
           token: ${{ secrets.CODSPEED_TOKEN }}
           run: pytest tests/ --codspeed

.github/workflows/docs-pr-close.yml

@@ -19,12 +19,12 @@ jobs:
           fetch-depth: 0
 
       - name: Setup Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python-version }}
+          python-version: "3.11"
 
       - name: Install dependencies
-        run: pip install -e .[docs]
+        run: ./scripts/install_docs_dependencies.sh
 
       - name: Set git credentials
         run: |

.github/workflows/docs-pr.yml

@@ -22,23 +22,19 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Setup Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python-version }}
-          # Looks like it's not working very well for other people:
-          # https://github.com/actions/setup-python/issues/436
-          # cache: "pip"
-          # cache-dependency-path: pyproject.toml
+          python-version: "3.11"
 
-      - uses: actions/cache@v3
+      - uses: actions/cache@v4
         id: cache
         with:
           path: ${{ env.pythonLocation }}
           key: ${{ runner.os }}-python-${{ env.pythonLocation }}-${{ hashFiles('pyproject.toml') }}-docs-pr-v00
 
       - name: Install dependencies
         if: steps.cache.outputs.cache-hit != 'true'
-        run: pip install -e .[docs]
+        run: ./scripts/install_docs_dependencies.sh
 
       - name: Set git credentials
         run: |

.github/workflows/docs.yml

@@ -24,23 +24,19 @@ jobs:
       - uses: actions/checkout@v4
 
       - name: Setup Python
-        uses: actions/setup-python@v4
+        uses: actions/setup-python@v5
         with:
-          python-version: ${{ matrix.python-version }}
-          # Looks like it's not working very well for other people:
-          # https://github.com/actions/setup-python/issues/436
-          # cache: "pip"
-          # cache-dependency-path: pyproject.toml
+          python-version: "3.11"
 
-      - uses: actions/cache@v3
+      - uses: actions/cache@v4
         id: cache
         with:
           path: ${{ env.pythonLocation }}
           key: ${{ runner.os }}-python-${{ env.pythonLocation }}-${{ hashFiles('pyproject.toml') }}-docs-v00
 
       - name: Install dependencies
         if: steps.cache.outputs.cache-hit != 'true'
-        run: pip install -e .[docs]
+        run: ./scripts/install_docs_dependencies.sh
 
       - name: Check no warnings
         run: mkdocs build --strict

examples/arena_hard.py

@@ -15,11 +15,12 @@
 import re
 from typing import Any, Dict, List, Optional, Union
 
+from typing_extensions import override
+
 from distilabel.steps import GlobalStep, StepInput
 from distilabel.steps.tasks.base import Task
 from distilabel.steps.tasks.typing import ChatType
 from distilabel.steps.typing import StepOutput
-from typing_extensions import override
 
 
 class ArenaHard(Task):

examples/deepseek_prover.py

@@ -17,14 +17,15 @@
 from textwrap import dedent
 from typing import Any, Dict, List, Optional, Union
 
+from jinja2 import Template
+from pydantic import PrivateAttr
+from typing_extensions import override
+
 from distilabel.llms import InferenceEndpointsLLM
 from distilabel.pipeline import Pipeline
 from distilabel.steps import LoadDataFromHub
 from distilabel.steps.tasks.base import Task
 from distilabel.steps.tasks.typing import ChatType
-from jinja2 import Template
-from pydantic import PrivateAttr
-from typing_extensions import override
 
 _PARSE_DEEPSEEK_PROVER_AUTOFORMAL_REGEX = r"```lean4(.*?)```"
 

examples/structured_generation_with_instructor.py

@@ -14,11 +14,12 @@
 
 from typing import List
 
+from pydantic import BaseModel, Field
+
 from distilabel.llms import MistralLLM
 from distilabel.pipeline import Pipeline
 from distilabel.steps import LoadDataFromDicts
 from distilabel.steps.tasks import TextGeneration
-from pydantic import BaseModel, Field
 
 
 class Node(BaseModel):

examples/structured_generation_with_outlines.py

@@ -15,12 +15,13 @@
 from enum import Enum
 from pathlib import Path
 
+from pydantic import BaseModel, StringConstraints, conint
+from typing_extensions import Annotated
+
 from distilabel.llms import LlamaCppLLM
 from distilabel.pipeline import Pipeline
 from distilabel.steps import LoadDataFromDicts
 from distilabel.steps.tasks import TextGeneration
-from pydantic import BaseModel, StringConstraints, conint
-from typing_extensions import Annotated
 
 
 class Weapon(str, Enum):

scripts/install_docs_dependencies.sh

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+set -e
+
+python_version=$(python -c "import sys; print(sys.version_info[:2])")
+
+python -m pip install uv
+
+uv pip install --system -e ".[docs]"