Skip to content

Conversation

@ysiraichi
Copy link
Collaborator

Cherry pick of #9698 into r2.9 release branch.

@ysiraichi
Copy link
Collaborator Author

The CI failure is being addressed in #9706.

@ysiraichi ysiraichi merged commit 95446d3 into r2.9 Nov 10, 2025
21 of 24 checks passed
ysiraichi added a commit that referenced this pull request Nov 10, 2025
This fixes the `torchax` CI failures (e.g. [#9705 action][1]) caused by
the [`flax` dependency][2], which pulls `jax` 0.8.0 (or newer) for some
Python versions.

**Key Changes:**

- Set upper bound `<0.8.0` in `pyproject.toml` optional dependency
section
- Install `torchax` before the test dependencies

[1]:
https://github.com/pytorch/xla/actions/runs/19229289257/job/54963835669?pr=9705
[2]:
https://github.com/pytorch/xla/blob/611a5cc2675133e6b159a2be8b07b65c44656f29/torchax/dev-requirements.txt#L5
rajkthakur pushed a commit to rajkthakur/xla that referenced this pull request Nov 10, 2025
rajkthakur pushed a commit to rajkthakur/xla that referenced this pull request Nov 10, 2025
This fixes the `torchax` CI failures (e.g. [pytorch#9705 action][1]) caused by
the [`flax` dependency][2], which pulls `jax` 0.8.0 (or newer) for some
Python versions.

**Key Changes:**

- Set upper bound `<0.8.0` in `pyproject.toml` optional dependency
section
- Install `torchax` before the test dependencies

[1]:
https://github.com/pytorch/xla/actions/runs/19229289257/job/54963835669?pr=9705
[2]:
https://github.com/pytorch/xla/blob/611a5cc2675133e6b159a2be8b07b65c44656f29/torchax/dev-requirements.txt#L5
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants