ci(*:skip): Migrate CI to uv

.github/actions/bootstrap/action.yml

@@ -10,20 +10,9 @@ inputs:
   setuptools-version:
     description: "Version of setuptools to be installed using pip"
     default: 82.0.0
-  poetry-version:
-    description: "Version of poetry to be installed using pip"
-    default: 2.3.2
-  poetry-skip:
-    description: "Option to skip the poetry installation"
-    required: false
-    default: "false"
   uv-version:
     description: "Version of uv to install"
     default: "0.10.7"
-  uv-skip:
-    description: "Option to skip the uv installation"
-    required: false
-    default: "true"
 outputs:
   python-version:
     description: "Version range or exact version of Python or PyPy"
@@ -34,9 +23,6 @@ outputs:
   setuptools-version:
     description: "Installed version of setuptools"
     value: ${{ steps.versions.outputs.setuptools-version }}
-  poetry-version:
-    description: "Requested version of poetry"
-    value: ${{ steps.versions.outputs.poetry-version }}
   uv-version:
     description: "Requested version of uv"
     value: ${{ steps.versions.outputs.uv-version }}
@@ -48,24 +34,18 @@ runs:
       with:
         python-version: ${{ inputs.python-version }}
     - name: Set up uv
-      if: ${{ inputs.uv-skip == 'false' }}
       uses: astral-sh/setup-uv@v7
       with:
         version: ${{ inputs.uv-version }}
     - name: Install build tools
       run: |
         python -m pip install -U pip==${{ inputs.pip-version }}
         python -m pip install -U setuptools==${{ inputs.setuptools-version }}
-        if ${{ inputs.poetry-skip == 'false' }}; then
-          python -m pip install -U poetry==${{ inputs.poetry-version }}
-          python -m poetry config virtualenvs.create false
-        fi
       shell: bash
     - id: versions
       shell: bash
       run: |
         echo "python-version=$(echo ${{ inputs.python-version }})" >> $GITHUB_OUTPUT
         echo "pip-version=$(echo ${{ inputs.pip-version }})" >> $GITHUB_OUTPUT
         echo "setuptools-version=$(echo ${{ inputs.setuptools-version }})" >> $GITHUB_OUTPUT
-        echo "poetry-version=$(echo ${{ inputs.poetry-version }})" >> $GITHUB_OUTPUT
         echo "uv-version=$(echo ${{ inputs.uv-version }})" >> $GITHUB_OUTPUT

.github/workflows/build-deploy-non-framework-docs.yml

@@ -27,31 +27,33 @@ jobs:
         run: ./dev/test-copyright.sh
       - name: Bootstrap
         uses: ./.github/actions/bootstrap
-        with:
-          uv-skip: "false"
       - name: Install pandoc
         run: sudo apt install pandoc
       - name: Install docs dependencies
         run: |
           cd framework
-          python -m poetry install
-          python -m pip install -e ../datasets
+          uv sync --locked --all-groups
+          uv pip install -e ../datasets
       - name: Install devtool dependencies
         run: |
           cd dev
-          uv sync --frozen
+          uv sync --locked
       - name: Update HTML theme options
         run: |
           cd dev
-          uv run python -m devtool.update_html_themes
+          uv run --no-sync python -m devtool.update_html_themes
       - name: Build baselines docs
-        run: ./dev/build-baseline-docs.sh
+        run: |
+          cd framework
+          uv run --no-sync ../dev/build-baseline-docs.sh
       - name: Build examples docs
         run: |
           cd dev
-          uv run python -m devtool.build_example_docs
+          uv run --no-sync python -m devtool.build_example_docs
       - name: Build datasets docs
-        run: ./datasets/dev/build-flwr-datasets-docs.sh
+        run: |
+          cd framework
+          uv run --no-sync ../datasets/dev/build-flwr-datasets-docs.sh
       - name: Deploy docs
         if: ${{ github.ref == 'refs/heads/main' && github.repository == 'adap/flower' && !github.event.pull_request.head.repo.fork }}
         env:

.github/workflows/datasets-e2e.yml

@@ -48,8 +48,6 @@ jobs:
         uses: ./.github/actions/bootstrap
         with:
           python-version: 3.10.19
-          uv-skip: "false"
-          poetry-skip: "true"
       - name: Free Disk Space (Ubuntu)
         uses: jlumbroso/free-disk-space@v1.3.1
         with:
@@ -65,6 +63,6 @@ jobs:
           sudo apt-get update
           sudo apt-get install -y ffmpeg
       - name: Install dependencies
-        run: uv sync --frozen
+        run: uv sync --locked
       - name: Run tests
-        run: uv run python -m unittest discover -p '*_test.py'
+        run: uv run --no-sync python -m unittest discover -p '*_test.py'

.github/workflows/datasets-test.yml

@@ -50,8 +50,6 @@ jobs:
         uses: ./.github/actions/bootstrap
         with:
           python-version: ${{ matrix.python }}
-          uv-skip: "false"
-          poetry-skip: "true"
       - name: Free Disk Space (Ubuntu)
         uses: jlumbroso/free-disk-space@v1.3.1
         with:
@@ -69,7 +67,7 @@ jobs:
       - name: Install dependencies
         run: |
           cd datasets
-          uv sync --frozen --all-extras
+          uv sync --locked --all-extras
       - name: Cache Hugging Face datasets
         uses: actions/cache@v3
         with:
@@ -81,11 +79,11 @@ jobs:
           cd datasets
           if [ -n "${{ secrets.HF_TOKEN }}" ]; then
             echo "Logging into Hugging Face..."
-            uv run hf auth login --token ${{ secrets.HF_TOKEN }}
+            uv run --no-sync hf auth login --token ${{ secrets.HF_TOKEN }}
           else
             echo "Skipping Hugging Face login stage (HF_TOKEN not set)"
           fi
       - name: Test (formatting + unit tests)
         run: |
           cd datasets
-          uv run ./dev/test.sh
+          uv run --no-sync ./dev/test.sh

.github/workflows/framework-devtools.yml

@@ -30,6 +30,8 @@ jobs:
       - name: Install dependencies (mandatory + optional)
         run: |
           cd framework
-          python -m poetry install
+          uv sync --locked --all-groups
       - name: Lint + Test (isort/black/mypy/pylint/pytest)
-        run: ./framework/dev/test-tool.sh
+        run: |
+          cd framework
+          uv run --no-sync ./dev/test-tool.sh

.github/workflows/framework-docs-update-translations.yml

@@ -31,20 +31,20 @@ jobs:
       - name: Install dependencies
         run: |
           cd framework
-          python -m poetry install
+          uv sync --locked --all-groups
 
       - name: Install pandoc
         uses: nikeee/setup-pandoc@v1
 
       - name: Update text and translations for all locales
         run: |
-          cd framework/docs
-          make update-text
-          for langDir in locales/*; do
+          cd framework
+          uv run --no-sync make -C docs update-text
+          for langDir in docs/locales/*; do
             if [ -d "$langDir" ]; then
               lang=$(basename $langDir)
               echo "Updating language $lang"
-              make update-lang lang=$lang
+              uv run --no-sync make -C docs update-lang lang=$lang
             fi
           done
 

.github/workflows/framework-docs.yml

@@ -43,7 +43,7 @@ jobs:
       - name: Install Flower Framework
         run: |
           cd framework
-          python -m poetry install
+          uv sync --locked --all-groups
       - name: Sync required docs build assets from main for release branches
         if: ${{ startsWith(github.ref, 'refs/heads/release/framework-') || (github.event_name == 'pull_request' && startsWith(github.base_ref, 'release/framework-')) || (github.event_name == 'release' && startsWith(github.ref, 'refs/tags/framework-')) }}
         run: |
@@ -87,7 +87,8 @@ jobs:
           DOC_VERSION: ${{ steps.doc_version.outputs.version }}
         run: |
           echo "Building docs for version: ${DOC_VERSION}"
-          ./framework/dev/build-docs.sh full
+          cd framework
+          uv run --no-sync ./dev/build-docs.sh full
       - name: Determine if this is the latest stable version
         if: ${{ github.repository == 'adap/flower' && ((startsWith(github.ref, 'refs/heads/release/framework-') && (github.event_name == 'push' || github.event_name == 'workflow_dispatch')) || (github.event_name == 'release' && startsWith(github.ref, 'refs/tags/framework-'))) }}
         id: latest_check
@@ -141,8 +142,9 @@ jobs:
             git checkout origin/main
           fi
 
-          python ./framework/docs/generate_docs_ui_metadata.py \
-            --docs-ui-config ./dev/docs-ui-config.yml \
-            --output ./framework/docs/build/docs-ui-metadata.json
+          cd framework
+          uv run --no-sync python ./docs/generate_docs_ui_metadata.py \
+            --docs-ui-config ../dev/docs-ui-config.yml \
+            --output ./docs/build/docs-ui-metadata.json
 
-          aws s3 cp --cache-control "no-cache" ./framework/docs/build/docs-ui-metadata.json s3://${{ secrets.DOCS_AWS_BUCKET_NAME }}/docs/framework/docs-ui-metadata.json
+          aws s3 cp --cache-control "no-cache" ./docs/build/docs-ui-metadata.json s3://${{ secrets.DOCS_AWS_BUCKET_NAME }}/docs/framework/docs-ui-metadata.json

.github/workflows/framework-e2e.yml

@@ -47,11 +47,15 @@ jobs:
       - name: Install dependencies
         run: |
           cd framework
-          python -m poetry install --all-extras
+          uv sync --locked --all-extras --all-groups
       - name: Build wheel
-        run: ./framework/dev/build.sh
+        run: |
+          cd framework
+          uv run --no-sync ./dev/build.sh
       - name: Test wheel
-        run: ./framework/dev/test-wheel.sh
+        run: |
+          cd framework
+          uv run --no-sync ./dev/test-wheel.sh
       - name: Upload wheel
         if: ${{ github.repository == 'adap/flower' && !github.event.pull_request.head.repo.fork && github.actor != 'dependabot[bot]' }}
         id: upload
@@ -104,7 +108,6 @@ jobs:
         uses: ./.github/actions/bootstrap
         with:
           python-version: ${{ matrix.python-version }}
-          poetry-skip: 'true'
       - name: Install Flower from repo
         if: ${{ needs.changes.outputs.framework == 'true' && (github.repository != 'adap/flower' || github.event.pull_request.head.repo.fork || github.actor == 'dependabot[bot]') }}
         working-directory: ./framework
@@ -282,8 +285,6 @@ jobs:
       - name: Bootstrap
         if: ${{ needs.changes.outputs.framework == 'true' }}
         uses: ./.github/actions/bootstrap
-        with:
-          poetry-skip: 'true'
       - name: Install dependencies
         if: ${{ needs.changes.outputs.framework == 'true' }}
         run: |
@@ -421,7 +422,6 @@ jobs:
         uses: ./.github/actions/bootstrap
         with:
           python-version: ${{ matrix.python-version }}
-          poetry-skip: 'true'
       - name: Install Flower from repo
         if: ${{ needs.changes.outputs.framework == 'true' && (github.repository != 'adap/flower' || github.event.pull_request.head.repo.fork || github.actor == 'dependabot[bot]') }}
         run: |
@@ -469,7 +469,6 @@ jobs:
         uses: ./.github/actions/bootstrap
         with:
           python-version: ${{ matrix.python-version }}
-          poetry-skip: 'true'
       - name: Install Flower from repo
         if: ${{ needs.changes.outputs.framework == 'true' && (github.repository != 'adap/flower' || github.event.pull_request.head.repo.fork || github.actor == 'dependabot[bot]') }}
         working-directory: ./framework

.github/workflows/framework-release-nightly.yml

@@ -36,8 +36,8 @@ jobs:
           echo "pip-version=${{ steps.bootstrap.outputs.pip-version }}" >> "$GITHUB_OUTPUT"
           echo "setuptools-version=${{ steps.bootstrap.outputs.setuptools-version }}" >> "$GITHUB_OUTPUT"
 
-          NAME=$(poetry version | awk {'print $1'})
-          VERSION=$(poetry version -s)
+          NAME=$(python -c "import tomllib; print(tomllib.load(open('pyproject.toml', 'rb'))['project']['name'])")
+          VERSION=$(python -c "import tomllib; print(tomllib.load(open('pyproject.toml', 'rb'))['project']['version'])")
           python dev/build-docker-image-matrix.py --flwr-version "${VERSION}" --matrix nightly --flwr-package "${NAME}" > matrix.json
           echo "matrix=$(cat matrix.json)" >> $GITHUB_OUTPUT
 

.github/workflows/framework-release.yml

@@ -48,7 +48,7 @@ jobs:
         curl $wheel_url --output dist/$wheel_name
         curl $tar_url --output dist/$tar_name
 
-        python -m poetry publish -u __token__ -p ${{ secrets.PYPI_TOKEN_RELEASE_FLWR }}
+        uv publish --token ${{ secrets.PYPI_TOKEN_RELEASE_FLWR }}
 
   parameters:
     if: ${{ github.repository == 'adap/flower' }}

.github/workflows/framework-test.yml

@@ -67,14 +67,20 @@ jobs:
       - name: Install dependencies
         if: ${{ needs.changes.outputs.framework == 'true' || needs.changes.outputs.ex_bench == 'true' }}
         run: |
-          cd framework 
-          python -m poetry install --all-extras
+          cd framework
+          uv sync --python=${{ matrix.python }} --locked --all-extras --all-groups
       - name: Check if protos need recompilation
         if: ${{ needs.changes.outputs.framework == 'true' }}
-        run: ./framework/dev/check-protos.sh
+        run: |
+          cd framework
+          uv run --no-sync ./dev/check-protos.sh
       - name: Lint + Test (isort/black/docformatter/mypy/pylint/flake8/pytest)
         if: ${{ needs.changes.outputs.framework == 'true' }}
-        run: ./framework/dev/test.sh
+        run: |
+          cd framework
+          uv run --no-sync ./dev/test.sh
       - name: Check benchmarks and examples
         if: ${{ needs.changes.outputs.ex_bench == 'true' }}
-        run: ./dev/test.sh
+        run: |
+          cd framework
+          uv run --no-sync ../dev/test.sh

.github/workflows/intelligence-docs.yml

@@ -29,17 +29,14 @@ jobs:
       - uses: actions/checkout@v5
       - name: Bootstrap
         uses: ./.github/actions/bootstrap
-        with:
-          uv-skip: "false"
-          poetry-skip: "true"
       - name: Install dependencies
         run: |
           cd dev
-          uv sync --frozen
+          uv sync --locked
       - name: Check docs format
         run: |
           cd dev
-          uv run docstrfmt --check ../intelligence/docs/source
+          uv run --no-sync docstrfmt --check ../intelligence/docs/source
   build_ts_docs:
     runs-on: ubuntu-22.04
     name: Build TypeScript Docs
@@ -146,7 +143,7 @@ jobs:
         if: ${{ github.ref == 'refs/heads/main' && github.repository == 'adap/flower' && !github.event.pull_request.head.repo.fork }}
         run: |
           cd framework
-          python -m poetry install --extras "simulation"
+          uv sync --locked --all-groups --extra simulation
       - name: Download TS docs
         if: ${{ github.ref == 'refs/heads/main' && github.repository == 'adap/flower' && !github.event.pull_request.head.repo.fork }}
         uses: actions/download-artifact@v4
@@ -168,8 +165,8 @@ jobs:
       - name: Build final docs
         if: ${{ github.ref == 'refs/heads/main' && github.repository == 'adap/flower' && !github.event.pull_request.head.repo.fork }}
         run: |
-          cd intelligence/docs
-          make html
+          cd framework
+          uv run --no-sync make -C ../intelligence/docs html
       - name: Deploy docs
         if: ${{ github.ref == 'refs/heads/main' && github.repository == 'adap/flower' && !github.event.pull_request.head.repo.fork }}
         env:

.github/workflows/repo-check-pr-title.yml

@@ -20,7 +20,6 @@ jobs:
         uses: ./.github/actions/bootstrap
         with:
           python-version: 3.13
-          poetry-skip: 'true'
       - name: Check PR title format 
         env:
           PR_TITLE: ${{ github.event.pull_request.title }}

dev/README.md

@@ -4,7 +4,7 @@ This directory provides developer tooling used across the repository (for exampl
 formatting, docs helpers, and repository checks).
 
 `devtool` uses `uv` for dependency management and command execution.
-The Flower `framework` project remains Poetry-based.
+The Flower `framework` project also uses `uv` for CI and local developer workflows.
 
 ## Prerequisites
 
@@ -72,4 +72,4 @@ uv sync --frozen
 
 - CI jobs that use `devtool` should run `uv sync --frozen` and execute commands with
   `uv run`.
-- CI jobs that install `framework` still use Poetry.
+- CI jobs that install `framework` should use `uv sync` and `uv run`.

framework/dev/build.sh

@@ -18,4 +18,4 @@
 set -e
 cd "$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"/../
 
-python -m poetry build
+uv build --clear

framework/dev/publish-nightly.sh

@@ -28,8 +28,8 @@ cd "$(cd "$(dirname "${BASH_SOURCE[0]}")" >/dev/null 2>&1 && pwd)"/../
 if [[ $(git log --since="24 hours ago" --pretty=oneline) ]]; then
     sed -i -E "s/^name = \"(.+)\"/name = \"\1-nightly\"/" pyproject.toml
     sed -i -E "s/^version = \"(.+)\"/version = \"\1.dev$(date '+%Y%m%d')\"/" pyproject.toml
-    python -m poetry build
-    python -m poetry publish -u __token__ -p $PYPI_TOKEN
+    uv build --clear
+    uv publish --token "$PYPI_TOKEN"
 else
     echo "There were no commits in the last 24 hours."
 fi