checksums; add missing patch

easybuild/easyconfigs/j/jax/jax-0.4.35-gfbf-2024a-CUDA-12.6.0.eb

@@ -94,7 +94,7 @@ components = [
         'patches': [
             'jax-0.4.35_easyblock_compat.patch',
             'jax-0.4.35_fix-pybind11-systemlib_cupti.patch',
-            'jax-v0.4.35_version.patch',
+            'jax-0.4.35_version.patch',
         ],
         'checksums': [
             {'jax-v0.4.35.tar.gz':
@@ -107,6 +107,8 @@ components = [
              'cbf4ad92b8438c4ce2a975efce1c47c57d4c3b117bceee071ab660f964057223'},
             {'jax-0.4.35_fix-pybind11-systemlib_cupti.patch':
              '78efe6b5108a5da1935258286c94dea8438fd03651533c34023eeba27f514130'},
+            {'jax-0.4.35_version.patch':
+             'cd2139a7802abf14b4b2cecee331aed80fff2ef91e16fa105093aea0795455e8'},
         ],
         'start_dir': 'jax-jax-v%(version)s',
         'buildopts': _jaxlib_buildopts,
@@ -148,9 +150,10 @@ exts_list = [
         'source_tmpl': '%(name)s-v%(version)s.tar.gz',
         'source_urls': ['https://github.com/google/jax/archive/'],
         # 'patches': ['jax-0.4.25_fix_env_test_no_log_spam.patch'],  # TODO: still required? update?
-        'patches': ['jax-v0.4.35_version.patch'],
+        'patches': ['jax-0.4.35_version.patch'],
         'checksums': [
             {'jax-v0.4.35.tar.gz': '65e086708ae56670676b7b2340ad82b901d8c9993d1241a839c8990bdb8d6212'},
+            {'jax-0.4.35_version.patch': 'cd2139a7802abf14b4b2cecee331aed80fff2ef91e16fa105093aea0795455e8'},
         ],
         'runtest': local_test,
         'preinstallopts': _no_devtag

easybuild/easyconfigs/j/jax/jax-0.4.35_version.patch

@@ -0,0 +1,19 @@
+diff -ru jax-jax-v0.4.35/jax/version.py jax-jax-v0.4.35_version/jax/version.py
+--- jax-jax-v0.4.35/jax/version.py	2024-10-22 21:00:23.000000000 +0200
++++ jax-jax-v0.4.35_version/jax/version.py	2024-11-28 13:10:52.508536023 +0100
+@@ -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
+   return _version_from_git_tree(_version) or _version_from_todays_date(_version)
+@@ -71,6 +72,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.environ.get('JAX_NIGHTLY') or os.environ.get('JAXLIB_NIGHTLY'):