Skip to content

Add torch 2.10.0 to unit tests and use python 3.14#3450

Merged
winglian merged 3 commits intomainfrom
ci-2100
Mar 3, 2026
Merged

Add torch 2.10.0 to unit tests and use python 3.14#3450
winglian merged 3 commits intomainfrom
ci-2100

Conversation

@winglian
Copy link
Collaborator

@winglian winglian commented Mar 3, 2026

Description

Motivation and Context

How has this been tested?

AI Usage Disclaimer

Screenshots (if appropriate)

Types of changes

Social Handles (Optional)

Summary by CodeRabbit

  • Chores
    • Updated CI/CD testing infrastructure with expanded version matrices for Python and PyTorch compatibility testing.
    • Enhanced Docker end-to-end testing configuration with additional CUDA and dependency variants.
    • Adjusted testing pipeline parameters for improved compatibility coverage.

@coderabbitai
Copy link
Contributor

coderabbitai bot commented Mar 3, 2026

Important

Review skipped

Auto incremental reviews are disabled on this repository.

Please check the settings in the CodeRabbit UI or the .coderabbit.yaml file in this repository. To trigger a single review, invoke the @coderabbitai review command.

You can disable this status message by setting the reviews.review_status to false in the CodeRabbit configuration file.

Use the checkbox below for a quick retry:

  • 🔍 Trigger review
📝 Walkthrough

Walkthrough

Updates CI workflow matrices in .github/workflows/tests.yml to adjust Python versions (3.11–3.12 to 3.12–3.14) and PyTorch versions (2.8.0, 2.9.0, 2.9.1 to 2.8.0, 2.9.1, 2.10.0), adds a docker-e2e CUDA 12.8.1 variant, and modifies exclusion rules accordingly.

Changes

Cohort / File(s) Summary
CI Workflow Matrices
.github/workflows/tests.yml
Broadened pytest and pytest-sdist job matrices from Python [3.11, 3.12] and PyTorch [2.8.0, 2.9.0, 2.9.1] to Python [3.12, 3.14] and PyTorch [2.8.0, 2.9.1, 2.10.0]; adjusted exclusion rules from 3.12 to 3.14. Added docker-e2e matrix include for CUDA 12.8.1 (cuda 128) with Python 3.11 and PyTorch 2.10.0. Updated docker-e2e-cleanup variant to use Python 3.11 instead of 3.12.

Estimated code review effort

🎯 2 (Simple) | ⏱️ ~8 minutes

Possibly related PRs

  • #3430: Modifies GitHub Actions workflow matrices to add PyTorch 2.10.0 and CUDA 12.8.1 test variants.
  • #3369: Adjusts Python and PyTorch version matrices in tests.yml CI configuration.
  • #3272: Adds CUDA 128 with cuda_version 12.8.1 and Python 3.11 variant entries to workflow matrices.

Suggested labels

ready to merge

Suggested reviewers

  • djsaunde
  • SalmanMohammadi
🚥 Pre-merge checks | ✅ 3
✅ Passed checks (3 passed)
Check name Status Explanation
Description Check ✅ Passed Check skipped - CodeRabbit’s high-level summary is enabled.
Title check ✅ Passed The title accurately reflects the main changes in the pull request: updating CI workflows to include PyTorch 2.10.0 and switching to Python 3.14.
Docstring Coverage ✅ Passed No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.

✏️ Tip: You can configure your own custom pre-merge checks in the settings.

✨ Finishing Touches
🧪 Generate unit tests (beta)
  • Create PR with unit tests
  • Post copyable unit tests in a comment
  • Commit unit tests in branch ci-2100

Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out.

❤️ Share

Comment @coderabbitai help to get the list of available commands and usage tips.

Copy link
Contributor

@coderabbitai coderabbitai bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Actionable comments posted: 1

🧹 Nitpick comments (1)
.github/workflows/tests.yml (1)

380-380: Consider documenting why cleanup is pinned to Python 3.11.

A short inline comment would make this intentional divergence clearer versus the newer unit-test Python lanes.

🤖 Prompt for AI Agents
Verify each finding against the current code and only fix it if needed.

In @.github/workflows/tests.yml at line 380, Add a brief inline comment
explaining why the cleanup job pins python_version: "3.11" (e.g., compatibility
with a specific tool/fixture or avoiding a regression in newer Python versions)
so reviewers understand this intentional divergence from the other unit-test
Python lanes; update the cleanup job’s python_version: "3.11" line to include
that short comment nearby referencing the reason.
🤖 Prompt for all review comments with AI agents
Verify each finding against the current code and only fix it if needed.

Nitpick comments:
In @.github/workflows/tests.yml:
- Line 380: Add a brief inline comment explaining why the cleanup job pins
python_version: "3.11" (e.g., compatibility with a specific tool/fixture or
avoiding a regression in newer Python versions) so reviewers understand this
intentional divergence from the other unit-test Python lanes; update the cleanup
job’s python_version: "3.11" line to include that short comment nearby
referencing the reason.

ℹ️ Review info

Configuration used: Path: .coderabbit.yaml

Review profile: CHILL

Plan: Pro

📥 Commits

Reviewing files that changed from the base of the PR and between 945c8ae and 720fc75.

📒 Files selected for processing (1)
  • .github/workflows/tests.yml

Comment on lines +57 to +63
python_version: ["3.12", "3.14"]
pytorch_version: ["2.8.0", "2.9.1", "2.10.0"]
exclude:
- python_version: "3.12"
- python_version: "3.14"
pytorch_version: "2.8.0"
- python_version: "3.12"
pytorch_version: "2.9.0"
- python_version: "3.14"
pytorch_version: "2.9.1"
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

⚠️ Potential issue | 🔴 Critical

🧩 Analysis chain

🌐 Web query:

PyTorch 2.10.0 official torchvision version compatibility matrix Python 3.12 3.14

💡 Result:

  • There is no “PyTorch 2.10.0” in the official torch↔torchvision compatibility tables (current stable table goes up to torch 2.9 / torchvision 0.24). [1]

Official torch ↔ torchvision ↔ Python compatibility (stable)

From the torchvision PyPI compatibility matrix: [1]

torch torchvision Supported Python
2.6 0.21 >=3.9, <=3.12
2.7 0.22 >=3.9, <=3.13
2.8 0.23 >=3.9, <=3.13
2.9 0.24 >=3.10, <=3.14
nightly nightly >=3.10, <=3.14

What this means for your targets

  • Python 3.12: use torch 2.6 + torchvision 0.21 (or newer rows that still include 3.12, e.g., torch 2.7/2.8/2.9). [1]
  • Python 3.14: officially covered starting at torch 2.9 + torchvision 0.24 (or nightly). [1]

Sources: torchvision official PyPI compatibility table [1].


🏁 Script executed:

# Check the actual workflow file to verify the mentioned lines
cd .github/workflows && head -80 tests.yml | tail -30

Repository: axolotl-ai-cloud/axolotl

Length of output: 1745


🏁 Script executed:

# Also check the wider context around the matrix definition
sed -n '50,70p' .github/workflows/tests.yml

Repository: axolotl-ai-cloud/axolotl

Length of output: 1745


Remove non-existent PyTorch 2.10.0 from the matrix and pin torchvision versions.

PyTorch 2.10.0 does not exist in official stable releases; the latest stable is 2.9. According to the official PyTorch compatibility matrix, use:

  • pytorch_version: ["2.8.0", "2.9.1"]
  • For Python 3.14 + PyTorch 2.9.1: pair with torchvision==0.24
  • For Python 3.12/3.13 + PyTorch 2.8.0: pair with torchvision==0.23
  • For Python 3.12/3.13 + PyTorch 2.9.1: pair with torchvision==0.24

Add torchvision pinning to the matrix to avoid resolver drift and ensure reproducible CI runs.

@codecov
Copy link

codecov bot commented Mar 3, 2026

Codecov Report

✅ All modified and coverable lines are covered by tests.

📢 Thoughts on this report? Let us know!

@winglian winglian merged commit 653f90b into main Mar 3, 2026
23 of 24 checks passed
@winglian winglian deleted the ci-2100 branch March 3, 2026 18:01
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant