taking a stab...

Author: maurycy

dev.py

@@ -1,6 +1,8 @@
 # A script for running pyperformance out of the repo in dev-mode.
 
 import os.path
+import shutil
+import subprocess
 import sys
 
 REPO_ROOT = os.path.dirname(os.path.abspath(__file__))
@@ -29,20 +31,35 @@ def ensure_venv_ready(venvroot=None, kind="dev", venvsdir=VENVS):
         readyfile = os.path.join(sys.prefix, "READY")
         isready = os.path.exists(readyfile)
     else:
-        import venv
-
         if not venvroot:
             venvroot = resolve_venv_root(kind, venvsdir)
         # Make sure the venv exists.
         readyfile = os.path.join(venvroot, "READY")
         isready = os.path.exists(readyfile)
         if not isready:
             relroot = os.path.relpath(venvroot)
-            if not os.path.exists(venvroot):
-                print(f"creating venv at {relroot}...")
+            uv = shutil.which("uv")
+            if not uv:
+                sys.exit(
+                    "ERROR: uv executable not found. "
+                    "Install uv from https://astral.sh/uv and retry."
+                )
+            if os.path.exists(venvroot):
+                print(f"uv env {relroot} not ready, re-creating...")
+                shutil.rmtree(venvroot)
             else:
-                print(f"venv {relroot} not ready, re-creating...")
-            venv.create(venvroot, with_pip=True, clear=True)
+                print(f"creating uv env at {relroot}...")
+            result = subprocess.run(
+                [
+                    uv,
+                    "venv",
+                    "--python",
+                    sys.executable,
+                    venvroot,
+                ]
+            )
+            if result.returncode != 0:
+                sys.exit("ERROR: uv venv creation failed")
         else:
             assert os.path.exists(os.path.join(venvroot, "pyvenv.cfg"))
         # Return the venv's Python executable.
@@ -52,18 +69,31 @@ def ensure_venv_ready(venvroot=None, kind="dev", venvsdir=VENVS):
 
     # Now make sure the venv has pyperformance installed.
     if not isready:
-        import subprocess
-
         relroot = os.path.relpath(venvroot)
-        print(f"venv {relroot} not ready, installing dependencies...")
+        print(f"uv env {relroot} not ready, installing dependencies...")
+        uv = shutil.which("uv")
+        if not uv:
+            sys.exit(
+                "ERROR: uv executable not found. "
+                "Install uv from https://astral.sh/uv and retry."
+            )
         proc = subprocess.run(
-            [python, "-m", "pip", "install", "--upgrade", "--editable", REPO_ROOT],
+            [
+                uv,
+                "pip",
+                "install",
+                "--python",
+                python,
+                "--upgrade",
+                "--editable",
+                f"{REPO_ROOT}[dev]",
+            ],
         )
         if proc.returncode != 0:
-            sys.exit("ERROR: install failed")
+            sys.exit("ERROR: uv pip install failed")
         with open(readyfile, "w"):
             pass
-        print("...venv {relroot} ready!")
+        print(f"...uv env {relroot} ready!")
 
     return venvroot, python
 

pyperformance/_venv.py

@@ -2,6 +2,7 @@
 
 import os
 import os.path
+import shutil
 import sys
 import types
 
@@ -111,11 +112,24 @@ def create_venv(
 ):
     """Create a new venv at the given root, optionally installing pip."""
     already_existed = os.path.exists(root)
-    if withpip:
-        args = ["-m", "venv", root]
+    uv = shutil.which("uv")
+    if not uv:
+        print("ERROR: uv executable not found. Install uv from https://astral.sh/uv.")
+        raise VenvCreationFailedError(root, 127, already_existed)
+
+    if isinstance(python, str) or python is None:
+        target_python = python or sys.executable
     else:
-        args = ["-m", "venv", "--without-pip", root]
-    ec, _, _ = _utils.run_python(*args, python=python, env=env)
+        try:
+            target_python = python.sys.executable
+        except AttributeError as exc:
+            raise TypeError(f"expected python str, got {python!r}") from exc
+
+    argv = [uv, "venv"]
+    if target_python:
+        argv.extend(["--python", target_python])
+    argv.append(root)
+    ec, _, _ = _utils.run_cmd(argv, env=env)
     if ec != 0:
         if cleanonfail and not already_existed:
             _utils.safe_rmtree(root)

pyperformance/requirements/requirements.txt

@@ -1,9 +1,5 @@
-#
-# This file is autogenerated by pip-compile with Python 3.11
-# by the following command:
-#
-#    pip-compile --output-file=pyperformance/requirements/requirements.txt requirements.in
-#
+# This file was autogenerated by uv via the following command:
+#    uv pip compile -o requirements.txt requirements.in
 packaging==23.1
     # via -r requirements.in
 psutil==5.9.5

pyperformance/tests/__init__.py

@@ -75,16 +75,16 @@ def cleanup():
         def cleanup():
             return None
 
-    run_cmd(
-        python or sys.executable,
-        "-m",
-        "venv",
-        root,
-        capture=not verbose,
-        onfail="raise",
-        verbose=verbose,
-    )
-    return root, _resolve_venv_python(root), cleanup
+    uv = shutil.which("uv")
+    if not uv:
+        raise RuntimeError("uv executable is required to provision test environments")
+    argv = [uv, "venv"]
+    if python:
+        argv.extend(["--python", python])
+    argv.append(root)
+    run_cmd(*argv, capture=not verbose, onfail="raise", verbose=verbose)
+    venv_root = os.path.realpath(root)
+    return venv_root, _resolve_venv_python(venv_root), cleanup
 
 
 class CleanupFile:

pyproject.toml

@@ -1,8 +1,7 @@
 # Update dependencies:
 #
-#  - python3 -m pip install --user --upgrade pip-tools build
 #  - git clean -fdx  # remove all untracked files!
-#  - (pip-compile --upgrade -o requirements.txt requirements.in)
+#  - uv pip compile --upgrade -o requirements.txt requirements.in
 #
 # Prepare a release:
 #
@@ -28,8 +27,8 @@
 #  - git tag VERSION
 #  - git push --tags
 #  - Remove untracked files/dirs: git clean -fdx
-#  - python -m build
-#  - twine upload dist/*
+#  - uv run python -m build
+#  - uvx twine upload dist/*
 #
 # After the release:
 #
@@ -65,8 +64,9 @@ classifiers = [
 ]
 dynamic = [ "version" ]
 dependencies = [
-  "packaging",
-  "pyperf",
+  "packaging==23.1",
+  "psutil==5.9.5",
+  "pyperf==2.9",
   "tomli; python_version<'3.11'",
 ]
 

requirements.in

@@ -12,7 +12,7 @@
 pyperf
 
 # for benchmark metadata:
-packaging
+packaging==23.1
 tomli; python_version < '3.11'
 
 
@@ -22,4 +22,4 @@ tomli; python_version < '3.11'
 # The list of optional dependencies is hardcoded in pyperformance/venv.py
 
 # XXX Do we still need this?
-psutil
+psutil==5.9.5