CI(deps): Update jax (cpu) version to 0.10.0 in pyproject.toml

[njzjz]

Summary by CodeRabbit

Release Notes

• Chores
  • Updated JAX dependency version constraints to support Python 3.10 and 3.11 with optimized version requirements for each Python version.

[codecov]

Codecov Report

✅ All modified and coverable lines are covered by tests.
✅ Project coverage is 82.46%. Comparing base (d67e75b) to head (f5cbd41).

Additional details and impacted files

@@           Coverage Diff           @@
##           master    #5442   +/-   ##
=======================================
  Coverage   82.46%   82.46%           
=======================================
  Files         830      830           
  Lines       88414    88414           
  Branches     4232     4233    +1     
=======================================
  Hits        72913    72913           
+ Misses      14215    14213    -2     
- Partials     1286     1288    +2     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.
• 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

[coderabbitai]

No actionable comments were generated in the recent review. 🎉

ℹ️ Recent review info

⚙️ Run configuration

Configuration used: Repository UI

Review profile: CHILL

Plan: Pro

Run ID: 7229dd9f-a4c4-43f4-ac8a-df1081ce2eb6

📥 Commits

Reviewing files that changed from the base of the PR and between d67e75b and f5cbd41.

📒 Files selected for processing (1)

• pyproject.toml

---

📝 Walkthrough

Walkthrough

The pin_jax_cpu dependency group in pyproject.toml is updated with version-conditional JAX constraints: jax>=0.6.2 for Python 3.10+ and jax==0.10.0 for Python 3.11+, replacing the previous single jax==0.5.0 pin.

Changes

JAX CPU Dependency Pinning

Layer / File(s)	Summary
JAX CPU version constraints pyproject.toml	The pin_jax_cpu dependency group replaces jax==0.5.0 (Python >= 3.10) with two conditional constraints: jax>=0.6.2 for Python >= 3.10 and jax==0.10.0 for Python >= 3.11.

Estimated code review effort

🎯 1 (Trivial) | ⏱️ ~3 minutes

Possibly related PRs

• deepmodeling/deepmd-kit#5436: Also modifies the pin_jax_cpu dependency group in pyproject.toml to manage JAX version constraints across Python versions.

Suggested labels

build

Suggested reviewers

• wanghan-iapcm

🚥 Pre-merge checks | ✅ 5

✅ Passed checks (5 passed)

Check name	Status	Explanation
Description Check	✅ Passed	Check skipped - CodeRabbit’s high-level summary is enabled.
Title check	✅ Passed	The title accurately describes the main change: updating JAX CPU version constraints in pyproject.toml from jax==0.5.0 to jax>=0.6.2 and jax==0.10.0 with Python version conditions.
Docstring Coverage	✅ Passed	No functions found in the changed files to evaluate docstring coverage. Skipping docstring coverage check.
Linked Issues check	✅ Passed	Check skipped because no linked issues were found for this pull request.
Out of Scope Changes check	✅ Passed	Check skipped because no linked issues were found for this pull request.

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

✨ Finishing Touches

🧪 Generate unit tests (beta)

• Create PR with unit tests

Tip

💬 Introducing Slack Agent: The best way for teams to turn conversations into code.

Slack Agent is built on CodeRabbit's deep understanding of your code, so your team can collaborate across the entire SDLC without losing context.

• Generate code and open pull requests
• Plan features and break down work
• Investigate incidents and troubleshoot customer tickets together
• Automate recurring tasks and respond to alerts with triggers
• Summarize progress and report instantly

Built for teams:

• Shared memory across your entire org—no repeating context
• Per-thread sandboxes to safely plan and execute work
• Governance built-in—scoped access, auditability, and budget controls

One agent for your entire SDLC. Right inside Slack.

👉 Get started

---

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

• X
• Mastodon
• Reddit
• LinkedIn

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

[Copilot]

Pull request overview

Updates the CI “pinned” dependency group for CPU JAX in pyproject.toml, adjusting version constraints based on Python version to support newer Python environments.

Changes:

• Updated pin_jax_cpu from a single fixed JAX version to Python-version-specific constraints.
• Introduced an exact JAX pin (==0.10.0) for Python >= 3.11 and a lower-bounded constraint (>=0.6.2) for Python >= 3.10.

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.