Skip to content

jax/0.8.1: Fix jax version returned to not show .dev #25791

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
SimonPinches wants to merge 4 commits into
base: develop
Choose a base branch
from
Open

jax/0.8.1: Fix jax version returned to not show .dev #25791

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
e448dbf
jax/0.8.1: Fix jax version returned to not show .dev
SimonPinches Apr 16, 2026
b2dcfad
jax/0.8.1: Fix style issues
SimonPinches Apr 16, 2026
49bc857
remove patch that's already applied in jax 0.8.1
boegel Jun 20, 2026
5c6d319
revert back to using local variables
boegel Jun 20, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 7 additions & 4 deletions easybuild/easyconfigs/j/jax/jax-0.8.1-gfbf-2025b.eb
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -87,14 +87,17 @@ _runtest_cmd = ' '.join([_runtest_cmd1, '||', _runtest_cmd1, '--last-failed']),

exts_list = [
(name, version, {
'patches': ['jax-0.8.1_jax-version-fix.patch'],
'prebuildopts': _local_prebuildopts,
'testinstall': True,
"runtest": _runtest_cmd,
'sources': [
{'source_urls': ['https://github.com/google/jax/archive/'], 'filename': '%(name)s-v%(version)s.tar.gz'}
],
'sources': [{
'source_urls': ['https://github.com/google/jax/archive/'],
'filename': '%(name)s-v%(version)s.tar.gz',
}],
'testinstall': True,
'checksums': [
{'jax-v0.8.1.tar.gz': '38882602112dadfd49a2c74868a0722574ae88e04646a96f32f8c36a7893c548'},
{'jax-0.8.1_jax-version-fix.patch': '0feae9caa48e758eb3605bf432c837ec4c72afb25a7e565ee272c0d9c805f75e'},
],
}),
]
Expand Down
21 changes: 21 additions & 0 deletions easybuild/easyconfigs/j/jax/jax-0.8.1_jax-version-fix.patch
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,21 @@
# Fix jax version to not show .dev
# S.D. Pinches, 15 April 2026
diff -ru jax-jax-v0.8.1/jax/version.py.orig jax-jax-v0.8.1/jax/version.py
--- jax-jax-v0.8.1/jax/version.py.orig 2025-04-01 12:00:00.000000000 +0200
+++ jax-jax-v0.8.1/jax/version.py 2026-04-15 12:00:00.000000000 +0200
@@ -33,6 +33,7 @@
def _get_version_string() -> str:
# The build/source distribution for jax & jaxlib overwrites _release_version.
# In this case we return it directly.
+ return _version
if _release_version is not None:
return _release_version
if os.getenv("WHEEL_VERSION_SUFFIX"):
@@ -82,6 +83,7 @@
- if JAX_NIGHTLY or JAXLIB_NIGHTLY are set: version looks like "0.4.16.dev20230906"
- if none are set: version looks like "0.4.16.dev20230906+ge58560fdc
"""
+ return _version
if _release_version is not None:
return _release_version
if os.getenv("WHEEL_VERSION_SUFFIX"):
Loading