chore: migrate from poetry/pip to uv for dependency management (#3327)

Co-authored-by: Copilot <223556219+Copilot@users.noreply.github.com>
Co-authored-by: Amp <amp@ampcode.com>

Author: 3 people

.github/workflows/ci.yml

@@ -34,20 +34,16 @@ jobs:
       - name: Apt install
         run: cat Aptfile | sudo xargs apt-get install
 
-      - name: Set up Python
-        uses: actions/setup-python@a309ff8b426b58ec0e2a45f0f869d46889d02405 # v6
+      - name: Install uv
+        uses: astral-sh/setup-uv@v7
         with:
-          python-version: "3.11"
+          enable-cache: true
 
-      - name: Install poetry
-        uses: snok/install-poetry@v1
-        with:
-          version: 2.1.3
-          virtualenvs-create: true
-          virtualenvs-in-project: true
+      - name: Install Python
+        run: uv python install 3.11
 
       - name: Install dependencies
-        run: poetry install --no-interaction
+        run: uv sync --locked --no-group prod
 
       - name: Create test local state
         run: ./scripts/test/stub-data.sh
@@ -69,7 +65,7 @@ jobs:
           OPENEDX_API_CLIENT_SECRET: fake_client_secret  # pragma: allowlist secret
 
       - name: Django system checks
-        run: poetry run ./manage.py check --fail-level WARNING
+        run: uv run ./manage.py check --fail-level WARNING
         env:
           CELERY_TASK_ALWAYS_EAGER: 'True'
           CELERY_BROKER_URL: redis://localhost:6379/4

AGENTS.md

@@ -21,25 +21,25 @@ MITx Online is a Django-based web platform for managing MIT online courses and p
 **Run tests:**
 ```bash
 # Full test suite with parallel execution
-poetry run pytest -n logical
+uv run pytest -n logical
 
 # Single test file
-poetry run pytest courses/api_test.py
+uv run pytest courses/api_test.py
 
 # Single test function
-poetry run pytest courses/api_test.py::test_function_name
+uv run pytest courses/api_test.py::test_function_name
 
 # With coverage
-poetry run pytest --cov . --cov-report html
+uv run pytest --cov . --cov-report html
 ```
 
 **Linting and formatting:**
 ```bash
 # Format code (ruff)
-poetry run ruff format .
+uv run ruff format .
 
 # Lint with auto-fix
-poetry run ruff check --fix .
+uv run ruff check --fix .
 
 # Pre-commit hooks (runs all checks)
 pre-commit run --all-files
@@ -199,7 +199,7 @@ course = CourseFactory.create()
 
 **Always check for missing migrations before committing:**
 ```bash
-poetry run python manage.py makemigrations --check --dry-run
+uv run python manage.py makemigrations --check --dry-run
 ```
 
 Test suite includes checks for:
@@ -250,9 +250,9 @@ These provide standard patterns used across MIT ODL projects.
 
 ### Dependency Management
 
-**Python:** Use Poetry for dependency management:
+**Python:** Use uv for dependency management:
 ```bash
-docker compose run --rm web poetry add <package>
+docker compose run --rm web uv add <package>
 docker compose build web celery  # Rebuild images after changes
 ```
 
@@ -298,16 +298,16 @@ Optional OIDC authentication via Keycloak:
 The test suite uses `pytest-django` with parallel execution via `pytest-xdist`:
 ```bash
 # Single test module
-poetry run pytest courses/models_test.py
+uv run pytest courses/models_test.py
 
 # Single test class
-poetry run pytest courses/models_test.py::TestCourse
+uv run pytest courses/models_test.py::TestCourse
 
 # Single test method
-poetry run pytest courses/models_test.py::TestCourse::test_course_creation
+uv run pytest courses/models_test.py::TestCourse::test_course_creation
 
 # With debugging (disables parallel execution)
-poetry run pytest -n0 -s courses/models_test.py::test_function
+uv run pytest -n0 -s courses/models_test.py::test_function
 ```
 
 ### Common Management Commands
@@ -331,7 +331,7 @@ manage.py collectstatic              # Collect static files
 Generate and view API schema:
 ```bash
 # Generate schema
-poetry run python manage.py spectacular --file schema.yml
+uv run python manage.py spectacular --file schema.yml
 
 # View in browser
 # Navigate to http://mitxonline.odl.local:8013/api/schema/swagger-ui/

Dockerfile

@@ -20,46 +20,36 @@ RUN mkdir /src && \
     adduser --disabled-password --gecos "" mitodl && \
     mkdir /var/media && chown -R mitodl:mitodl /var/media
 
-FROM system AS poetry
+FROM system AS uv
 
 # copy in trusted certs
 COPY --chmod=644 certs/*.crt /usr/local/share/ca-certificates/
 RUN update-ca-certificates
 
-# Poetry env configuration
+# uv env configuration
 ENV  \
-  # poetry:
   PYTHONUNBUFFERED=1 \
   PYTHONDONTWRITEBYTECODE=1 \
-  PIP_DISABLE_PIP_VERSION_CHECK=on \
-  POETRY_VERSION=2.1.3 \
-  POETRY_VIRTUALENVS_CREATE=false \
-  POETRY_CACHE_DIR='/tmp/cache/poetry' \
-  POETRY_HOME='/home/mitodl/.local' \
-  VIRTUAL_ENV="/opt/venv"
-ENV PATH="$VIRTUAL_ENV/bin:$POETRY_HOME/bin:$PATH"
+  UV_PROJECT_ENVIRONMENT="/opt/venv"
+ENV PATH="/opt/venv/bin:$PATH"
 
-RUN pip install --no-cache-dir "poetry==$POETRY_VERSION"
+# Install uv
+COPY --from=ghcr.io/astral-sh/uv:latest /uv /uvx /usr/local/bin/
 
 COPY pyproject.toml /src
-COPY poetry.lock /src
+COPY uv.lock /src
 COPY mitol_*.gz /src
 
 RUN chown -R mitodl:mitodl /src && \
-    mkdir ${VIRTUAL_ENV} && chown -R mitodl:mitodl ${VIRTUAL_ENV}
+    mkdir -p /opt/venv && \
+    chown -R mitodl:mitodl /opt/venv
 
 USER mitodl
-RUN curl -sSL https://install.python-poetry.org \
-  | \
-  POETRY_VERSION=${POETRY_VERSION} \
-  POETRY_HOME=${POETRY_HOME} \
-  python3 -q
 WORKDIR /src
-RUN python3 -m venv $VIRTUAL_ENV
-RUN poetry install
+RUN uv sync --frozen --no-install-project --group prod
 
 
-FROM poetry AS code
+FROM uv AS code
 
 COPY . /src
 WORKDIR /src
@@ -90,6 +80,6 @@ COPY --from=node /src /src
 
 FROM code AS jupyter-notebook
 
-RUN pip install --force-reinstall jupyter
+RUN uv pip install --force-reinstall jupyter
 
 USER mitodl

README.md

@@ -44,13 +44,13 @@ In order to manage the certificates, follow these steps:
 
 # Updating python dependencies
 
-Python dependencies are managed with poetry.  If you need to add a new dependency, run this command:
+Python dependencies are managed with uv.  If you need to add a new dependency, run this command:
 
 ```
-docker compose run --rm web poetry add <dependency>
+docker compose run --rm web uv add <dependency>
 ```
-This will update the `pyproject.toml` and `poetry.lock` files.  Then run `docker-compose build web celery` to make the change permanent in your docker images.
-Refer to the [poetry documentation](https://python-poetry.org/docs/cli/) for particulars about specifying versions, removing dependencies, etc.
+This will update the `pyproject.toml` and `uv.lock` files.  Then run `docker-compose build web celery` to make the change permanent in your docker images.
+Refer to the [uv documentation](https://docs.astral.sh/uv/reference/cli/) for particulars about specifying versions, removing dependencies, etc.
 
 
 # Generating documentation

b2b/api_test.py

@@ -281,7 +281,7 @@ def test_ensure_enrollment_codes(  # noqa: PLR0913
     mocker.patch("openedx.tasks.clone_courserun.delay")
     mocked_ensure_call = mocker.patch("b2b.tasks.queue_enrollment_code_check.delay")
     max_learners = FAKE.random_int(min=1, max=15) if has_learner_cap else None
-    price = FAKE.random_int(min=0, max=100) if has_price else None
+    price = FAKE.random_int(min=1, max=100) if has_price else None
     assert_price = price if price else Decimal(0)
 
     contract = factories.ContractPageFactory(
@@ -326,7 +326,7 @@ def test_ensure_enrollment_codes(  # noqa: PLR0913
     #   no longer have discounts.
     if update_no_price or update_sso or update_change_price:
         if update_change_price:
-            price = FAKE.random_int(min=0, max=100)
+            price = FAKE.random_int(min=1, max=100)
             assert_price = price if price else Decimal(0)
             contract.enrollment_fixed_price = price
         if update_no_price:

docs/source/configuration/uwsgi_tuning.rst

@@ -16,7 +16,7 @@ To set up locally:
 
 Set up uwsgitop
 ---------------
-1. Install uwsgitop: ``docker compose run --rm web poetry add uwsgitop``
+1. Install uwsgitop: ``docker compose run --rm web uv add uwsgitop``
 2. Set UWSGI_RELOAD_ON_RSS in your .env to a high value (e.g. 500)
 3. Set UWSGI_MAX_REQUESTS in your .env to a high value (e.g. 10000)
 4. ``docker compose build``
@@ -27,7 +27,7 @@ Set up uwsgitop
 
 Set up Locust
 -------------
-1. Install Locust: ``docker compose run --rm web poetry add locust``
+1. Install Locust: ``docker compose run --rm web uv add locust``
 2. Add locust to your docker-compose.yml locally, under services:
 
 .. code-block:: shell