support Python 3.14 (#899)

Author: skshetry

.github/workflows/release.yml

@@ -17,25 +17,21 @@ jobs:
     runs-on: ubuntu-latest
     steps:
       - name: Check out the repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           fetch-depth: 0
 
-      - name: Set up Python 3.10
-        uses: actions/setup-python@v5
+      - name: Set up Python 3.14
+        uses: actions/setup-python@v6
         with:
-          python-version: '3.10'
+          python-version: '3.14'
 
-      - name: Upgrade pip
-        run: |
-          pip install --upgrade pip
-          pip --version
-
-      - name: Install
-        run: python -m pip install build setuptools
+      - uses: astral-sh/setup-uv@v7
+      - name: Install nox
+        run: uv pip install --system nox --upgrade
 
       - name: Build package
-        run: python -m build
+        run: nox -s build
 
       - name: Upload package
         if: github.event_name == 'release'

.github/workflows/tests.yml

@@ -14,88 +14,81 @@ concurrency:
   cancel-in-progress: true
 
 jobs:
-  test_full:
-    timeout-minutes: 30
+  lint:
     runs-on: ubuntu-latest
-
     steps:
       - name: Check out the repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           fetch-depth: 0
 
       - name: Set up Python
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
-          python-version: "3.12"
+          python-version: "3.13"
 
-      - uses: astral-sh/setup-uv@v6
+      - uses: astral-sh/setup-uv@v7
         with:
-          enable-cache: true
-          cache-suffix: ${{ matrix.pyv }}
-          cache-dependency-glob: pyproject.toml
+          enable-cache: false
 
-      - name: Full install
-        run: uv pip install -e '.[dev]' --system
+      - name: Install nox
+        run: uv pip install --system nox --upgrade
 
       - uses: actions/cache@v4
         with:
           path: ~/.cache/pre-commit/
           key: pre-commit-4|${{ env.pythonLocation }}|${{ hashFiles('.pre-commit-config.yaml') }}
 
-      - name: pre-commit
-        run: uv tool run pre-commit run --show-diff-on-failure --color=always --all-files
-
-      - name: mypy
-        run: mypy
-
-      - name: Run tests
-        run: pytest -v tests --cov --cov-report=xml --cov-config=pyproject.toml
-
-      - name: Upload coverage report
-        uses: codecov/codecov-action@v5
-        with:
-          token: ${{ secrets.CODECOV_TOKEN }}
-          files: coverage.xml
-          flags: dvclive
+      - name: Lint code
+        run: nox -s lint
 
-  test_core:
+  tests:
     timeout-minutes: 30
-    runs-on: ${{ matrix.os }}
     strategy:
       fail-fast: false
       matrix:
         os: [ubuntu-latest, windows-latest, macos-latest]
-        pyv: ["3.9", "3.10", "3.11", "3.12"]
-
+        pyv: ["3.9", "3.10", "3.11", "3.12", "3.13", "3.14"]
+        type: [core_tests]
+        include:
+          - os: ubuntu-latest
+            pyv: "3.13"
+            type: tests
+        exclude:
+          - os: ubuntu-latest
+            pyv: "3.13"
+            type: core_tests
+    runs-on: ${{ matrix.os }}
     steps:
       - name: Check out the repository
-        uses: actions/checkout@v4
+        uses: actions/checkout@v5
         with:
           fetch-depth: 0
 
       - name: Set up Python ${{ matrix.pyv }}
-        uses: actions/setup-python@v5
+        uses: actions/setup-python@v6
         with:
           python-version: ${{ matrix.pyv }}
-          cache: "pip"
-          cache-dependency-path: setup.cfg
 
-      - name: Upgrade pip
-        run: |
-          python -m pip install --upgrade pip wheel
-          pip --version
+      - uses: astral-sh/setup-uv@v7
+        with:
+          enable-cache: false
 
-      - name: Install core
-        run: |
-          pip install -e '.[tests]'
+      - name: Install nox
+        run: uv pip install --system nox --upgrade
 
       - name: Run tests
-        run: pytest -v tests --ignore=tests/frameworks
+        run: nox -s ${{ matrix.type }}-${{ matrix.pyv }} -- --cov-report=xml
+
+      - name: Build package
+        run: nox -s build
+
+      - name: Upload coverage report
+        uses: codecov/codecov-action@v5
 
   check:
     if: always()
-    needs: [test_full, test_core]
+    needs: [lint, tests]
     runs-on: ubuntu-latest
     steps:
       - uses: re-actors/alls-green@release/v1

noxfile.py

@@ -5,16 +5,32 @@
 
 import nox
 
+nox.options.default_venv_backend = "uv|virtualenv"
 nox.options.reuse_existing_virtualenvs = True
 nox.options.sessions = "lint", "tests"
-locations = "src", "tests"
 
+project = nox.project.load_toml()
+python_versions = nox.project.python_versions(project)
 
-@nox.session(python=["3.9", "3.10", "3.11", "3.12", "pypy3.9", "pypy3.10"])
+
+@nox.session(python=python_versions)
 def tests(session: nox.Session) -> None:
+    session.install(".[dev]")
+    session.run(
+        "pytest",
+        "--cov",
+        "--cov-config=pyproject.toml",
+        *session.posargs,
+        env={"COVERAGE_FILE": f".coverage.{session.python}"},
+    )
+
+
+@nox.session(python=python_versions)
+def core_tests(session: nox.Session) -> None:
     session.install(".[tests]")
     session.run(
         "pytest",
+        "--ignore=tests/frameworks",
         "--cov",
         "--cov-config=pyproject.toml",
         *session.posargs,
@@ -42,8 +58,8 @@ def safety(session: nox.Session) -> None:
 
 @nox.session
 def build(session: nox.Session) -> None:
-    session.install("build", "setuptools", "twine")
-    session.run("python", "-m", "build")
+    session.install("twine", "uv")
+    session.run("uv", "build")
     dists = glob.glob("dist/*")
     session.run("twine", "check", *dists, silent=True)
 

pyproject.toml

@@ -3,8 +3,8 @@ requires = ["setuptools>=77", "setuptools_scm[toml]>=8"]
 build-backend = "setuptools.build_meta"
 
 [project]
-description = "Experiments logger for ML projects."
 name = "dvclive"
+description = "Experiments logger for ML projects."
 readme = "README.md"
 keywords = [
   "ai",
@@ -18,6 +18,7 @@ keywords = [
   "reproducibility"
 ]
 license = "Apache-2.0"
+license-files = ["LICENSE"]
 maintainers = [{name = "Iterative", email = "[email protected]"}]
 authors = [{name = "Iterative", email = "[email protected]"}]
 requires-python = ">=3.9"
@@ -27,7 +28,9 @@ classifiers = [
   "Programming Language :: Python :: 3.9",
   "Programming Language :: Python :: 3.10",
   "Programming Language :: Python :: 3.11",
-  "Programming Language :: Python :: 3.12"
+  "Programming Language :: Python :: 3.12",
+  "Programming Language :: Python :: 3.13",
+  "Programming Language :: Python :: 3.14"
 ]
 dynamic = ["version"]
 dependencies = [
@@ -59,11 +62,6 @@ tests = [
   "transformers[torch]",
   "tf-keras"
 ]
-dev = [
-  "dvclive[all,tests]",
-  "mypy==1.17.1",
-  "types-PyYAML"
-]
 mmcv = ["mmcv"]
 tf = ["tensorflow"]
 xgb = ["xgboost"]
@@ -75,6 +73,11 @@ optuna = ["optuna"]
 all = [
   "dvclive[image,mmcv,tf,xgb,lgbm,huggingface,fastai,lightning,optuna,plots,markdown]"
 ]
+dev = [
+  "dvclive[image,tf,xgb,lgbm,huggingface,fastai,lightning,optuna,plots,markdown,tests]",
+  "mypy==1.17.1",
+  "types-PyYAML"
+]
 
 [project.urls]
 Homepage = "https://github.com/iterative/dvclive"