Migrate from Poetry to uv

Author: BrianMichell

.github/workflows/constraints-poetry.txt

@@ -1,4 +1,2 @@
-pip==25.0.1
+bump-my-version==2024.1130
 nox==2025.2.9
-nox-poetry==1.2.0
-virtualenv==20.29.2

.github/workflows/constraints.txt

@@ -16,20 +16,16 @@ jobs:
         with:
           fetch-depth: 2
 
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install the latest version of uv
+        uses: astral-sh/setup-uv@v5
         with:
-          python-version: "3.12"
+          python-version: 3.13
+          pyproject-file: "${{ github.workspace }}/pyproject.toml"
 
-      - name: Upgrade pip
+      - name: Install bumpversion
         run: |
-          pip install --constraint=.github/workflows/constraints.txt pip
-          pip --version
-
-      - name: Install Poetry
-        run: |
-          pip install --constraint=.github/workflows/constraints-poetry.txt poetry
-          poetry --version
+          uv tool install --constraint=.github/workflows/constraints.txt bump-my-version
+          bump-my-version --version
 
       - name: Check if there is a parent commit
         id: check-parent-commit
@@ -42,18 +38,18 @@ jobs:
         uses: salsify/[email protected]
         with:
           version-command: |
-            bash -o pipefail -c "poetry version | awk '{ print \$2 }'"
+            bump-my-version show current_version
 
       - name: Bump version for developmental release
         if: "! steps.check-version.outputs.tag"
         run: |
-          poetry version patch &&
-          version=$(poetry version | awk '{ print $2 }') &&
-          poetry version $version.dev.$(date +%s)
+          bump-my-version bump patch &&
+          version=$(bump-my-version show current_version) &&
+          bump-my-version bump --new-version $version.dev$(date +%s)
 
       - name: Build package
         run: |
-          poetry build --ansi
+          uv build
 
       - name: Publish package on PyPI
         if: steps.check-version.outputs.tag

.github/workflows/release.yml

@@ -42,30 +42,15 @@ jobs:
         with:
           python-version: ${{ matrix.python }}
 
-      - name: Upgrade pip
-        run: |
-          pip install -c ${{ github.workspace }}/.github/workflows/constraints.txt pip
-          pip --version
-
-      - name: Upgrade pip in virtual environments
-        shell: python
-        run: |
-          import os
-          import pip
-
-          with open(os.environ["GITHUB_ENV"], mode="a") as io:
-              print(f"VIRTUALENV_PIP={pip.__version__}", file=io)
-
-      - name: Install Poetry
-        run: |
-          pipx install --pip-args="-c ${{ github.workspace }}/.github/workflows/constraints-poetry.txt" poetry
-          pipx inject poetry poetry-plugin-export
-          poetry --version
+       - name: Install the latest version of uv
+        uses: astral-sh/setup-uv@v5
+        with:
+          python-version: ${{ matrix.python }}
+          pyproject-file: ${{ github.workspace }}/pyproject.toml
 
       - name: Install Nox
         run: |
-          pipx install --pip-args="-c ${{ github.workspace }}/.github/workflows/constraints.txt" nox
-          pipx inject --pip-args="-c ${{ github.workspace }}/.github/workflows/constraints.txt" nox nox-poetry
+          uv tool install -c "${{ github.workspace }}/.github/workflows/constraints.txt" nox
           nox --version
 
       - name: Compute pre-commit cache key
@@ -118,26 +103,15 @@ jobs:
       - name: Check out the repository
         uses: actions/checkout@v4
 
-      - name: Set up Python
-        uses: actions/setup-python@v5
+      - name: Install the latest version of uv
+        uses: astral-sh/setup-uv@v5
         with:
-          python-version: "3.12"
-
-      - name: Upgrade pip
-        run: |
-          pip install -c ${{ github.workspace }}/.github/workflows/constraints.txt pip
-          pip --version
-
-      - name: Install Poetry
-        run: |
-          pipx install --pip-args="-c ${{ github.workspace }}/.github/workflows/constraints-poetry.txt" poetry
-          pipx inject poetry poetry-plugin-export
-          poetry --version
+          python-version: 3.13
+          pyproject-file: "${{ github.workspace }}/pyproject.toml"
 
       - name: Install Nox
         run: |
-          pipx install --pip-args="-c ${{ github.workspace }}/.github/workflows/constraints.txt" nox
-          pipx inject --pip-args="-c ${{ github.workspace }}/.github/workflows/constraints.txt" nox nox-poetry
+          uv tool install -c "${{ github.workspace }}/.github/workflows/constraints.txt" nox
           nox --version
 
       - name: Download coverage data

.github/workflows/tests.yml

@@ -6,21 +6,11 @@
 import sys
 from pathlib import Path
 from textwrap import dedent
+from tempfile import NamedTemporaryFile
 
 import nox
-
-
-try:
-    from nox_poetry import Session
-    from nox_poetry import session
-except ImportError:
-    message = f"""\
-    Nox failed to import the 'nox-poetry' package.
-
-    Please install it using the following command:
-
-    {sys.executable} -m pip install nox-poetry"""
-    raise SystemExit(dedent(message)) from None
+from nox import Session
+from nox import session
 
 
 package = "mdio"
@@ -37,80 +27,6 @@
 )
 
 
-def activate_virtualenv_in_precommit_hooks(session: Session) -> None:
-    """Activate virtualenv in hooks installed by pre-commit.
-
-    This function patches git hooks installed by pre-commit to activate the
-    session's virtual environment. This allows pre-commit to locate hooks in
-    that environment when invoked from git.
-
-    Args:
-        session: The Session object.
-    """
-    assert session.bin is not None  # noqa: S101
-
-    # Only patch hooks containing a reference to this session's bindir. Support
-    # quoting rules for Python and bash, but strip the outermost quotes so we
-    # can detect paths within the bindir, like <bindir>/python.
-    bindirs = [
-        bindir[1:-1] if bindir[0] in "'\"" else bindir
-        for bindir in (repr(session.bin), shlex.quote(session.bin))
-    ]
-
-    virtualenv = session.env.get("VIRTUAL_ENV")
-    if virtualenv is None:
-        return
-
-    headers = {
-        # pre-commit < 2.16.0
-        "python": f"""\
-            import os
-            os.environ["VIRTUAL_ENV"] = {virtualenv!r}
-            os.environ["PATH"] = os.pathsep.join((
-                {session.bin!r},
-                os.environ.get("PATH", ""),
-            ))
-            """,
-        # pre-commit >= 2.16.0
-        "bash": f"""\
-            VIRTUAL_ENV={shlex.quote(virtualenv)}
-            PATH={shlex.quote(session.bin)}"{os.pathsep}$PATH"
-            """,
-        # pre-commit >= 2.17.0 on Windows forces sh shebang
-        "/bin/sh": f"""\
-            VIRTUAL_ENV={shlex.quote(virtualenv)}
-            PATH={shlex.quote(session.bin)}"{os.pathsep}$PATH"
-            """,
-    }
-
-    hookdir = Path(".git") / "hooks"
-    if not hookdir.is_dir():
-        return
-
-    for hook in hookdir.iterdir():
-        if hook.name.endswith(".sample") or not hook.is_file():
-            continue
-
-        if not hook.read_bytes().startswith(b"#!"):
-            continue
-
-        text = hook.read_text()
-
-        if not any(
-            Path("A") == Path("a") and bindir.lower() in text.lower() or bindir in text
-            for bindir in bindirs
-        ):
-            continue
-
-        lines = text.splitlines()
-
-        for executable, header in headers.items():
-            if executable in lines[0].lower():
-                lines.insert(1, dedent(header))
-                hook.write_text("\n".join(lines))
-                break
-
-
 @session(name="pre-commit", python=python_versions[0])
 def precommit(session: Session) -> None:
     """Lint using pre-commit."""
@@ -120,7 +36,8 @@ def precommit(session: Session) -> None:
         "--hook-stage=manual",
         "--show-diff-on-failure",
     ]
-    session.install(
+    session.run(
+        "uv", "pip", "install",
         "black",
         "darglint",
         "flake8",
@@ -133,34 +50,45 @@ def precommit(session: Session) -> None:
         "pre-commit",
         "pre-commit-hooks",
         "pyupgrade",
+        external=True
     )
     session.run("pre-commit", *args)
-    if args and args[0] == "install":
-        activate_virtualenv_in_precommit_hooks(session)
 
 
 @session(python=python_versions[0])
 def safety(session: Session) -> None:
     """Scan dependencies for insecure packages."""
-    requirements = session.poetry.export_requirements()
-    session.install("safety")
-    # TODO(Altay): Remove the CVE ignore once its resolved. Its not critical, so ignoring now.
-    ignore = ["70612"]
-    session.run(
-        "safety",
-        "check",
-        "--full-report",
-        f"--file={requirements}",
-        f"--ignore={','.join(ignore)}",
-    )
+    with NamedTemporaryFile(delete=False) as requirements:
+        session.run(
+            "uv", "pip", "compile",
+            "pyproject.toml",
+            "--output-file", requirements.name,
+            external=True
+        )
+        session.run(
+            "uv", "pip", "install", "safety",
+            external=True
+        )
+        # TODO(Altay): Remove the CVE ignore once its resolved. Its not critical, so ignoring now.
+        ignore = ["70612"]
+        try:
+            session.run(
+                "safety",
+                "check",
+                "--full-report",
+                f"--file={requirements.name}",
+                f"--ignore={','.join(ignore)}",
+            )
+        finally:
+            os.remove(requirements.name)
 
 
 @session(python=python_versions)
 def mypy(session: Session) -> None:
     """Type-check using mypy."""
     args = session.posargs or ["src", "tests", "docs/conf.py"]
-    session.install(".")
-    session.install("mypy", "pytest")
+    session.run_always("uv", "pip", "install", "-e", ".", external=True)
+    session.run("uv", "pip", "install", "mypy", "pytest", external=True)
     session.run("mypy", *args)
     if not session.posargs:
         session.run("mypy", f"--python-executable={sys.executable}", "noxfile.py")
@@ -169,8 +97,12 @@ def mypy(session: Session) -> None:
 @session(python=python_versions)
 def tests(session: Session) -> None:
     """Run the test suite."""
-    session.install(".")
-    session.install("coverage[toml]", "pytest", "pygments", "pytest-dependency", "s3fs")
+    session.run_always("uv", "pip", "install", "-e", ".[cloud]", external=True)
+    session.run(
+        "uv", "pip", "install",
+        "coverage[toml]", "pytest", "pygments", "pytest-dependency", "s3fs",
+        external=True
+    )
     try:
         session.run("coverage", "run", "--parallel", "-m", "pytest", *session.posargs)
     finally:
@@ -183,7 +115,7 @@ def coverage(session: Session) -> None:
     """Produce the coverage report."""
     args = session.posargs or ["report"]
 
-    session.install("coverage[toml]")
+    session.run("uv", "pip", "install", "coverage[toml]", external=True)
 
     if not session.posargs and any(Path().glob(".coverage.*")):
         session.run("coverage", "combine")
@@ -194,8 +126,8 @@ def coverage(session: Session) -> None:
 @session(python=python_versions[0])
 def typeguard(session: Session) -> None:
     """Runtime type checking using Typeguard."""
-    session.install(".")
-    session.install("pytest", "typeguard", "pygments")
+    session.run_always("uv", "pip", "install", "-e", ".", external=True)
+    session.run("uv", "pip", "install", "pytest", "typeguard", "pygments", external=True)
     session.run("pytest", f"--typeguard-packages={package}", *session.posargs)
 
 
@@ -209,8 +141,8 @@ def xdoctest(session: Session) -> None:
         if "FORCE_COLOR" in os.environ:
             args.append("--colored=1")
 
-    session.install(".")
-    session.install("xdoctest[colors]")
+    session.run_always("uv", "pip", "install", "-e", ".", external=True)
+    session.run("uv", "pip", "install", "xdoctest[colors]", external=True)
     session.run("python", "-m", "xdoctest", *args)
 
 
@@ -221,14 +153,16 @@ def docs_build(session: Session) -> None:
     if not session.posargs and "FORCE_COLOR" in os.environ:
         args.insert(0, "--color")
 
-    session.install(".")
-    session.install(
+    session.run_always("uv", "pip", "install", "-e", ".", external=True)
+    session.run(
+        "uv", "pip", "install",
         "sphinx",
         "sphinx-click",
         "sphinx-copybutton",
         "furo",
         "myst-nb",
         "linkify-it-py",
+        external=True
     )
 
     build_dir = Path("docs", "_build")
@@ -242,15 +176,17 @@ def docs_build(session: Session) -> None:
 def docs(session: Session) -> None:
     """Build and serve the documentation with live reloading on file changes."""
     args = session.posargs or ["--open-browser", "docs", "docs/_build"]
-    session.install(".")
-    session.install(
+    session.run_always("uv", "pip", "install", "-e", ".", external=True)
+    session.run(
+        "uv", "pip", "install",
         "sphinx",
         "sphinx-autobuild",
         "sphinx-click",
         "sphinx-copybutton",
         "furo",
         "myst-nb",
         "linkify-it-py",
+        external=True
     )
 
     build_dir = Path("docs", "_build")