fix: unpin h5py on aarch64 (#4176)

Fix #3864.

<!-- This is an auto-generated comment: release notes by coderabbit.ai
-->

## Summary by CodeRabbit

- **New Features**
- Enhanced logic for detecting TensorFlow installation paths and
requirements.
	- Improved error handling for TensorFlow version detection.

- **Bug Fixes**
- Adjusted compatibility specifications for the `h5py` dependency on
aarch64 architecture.

- **Documentation**
- Updated comments to clarify changes in TensorFlow detection and
compatibility issues.

<!-- end of auto-generated comment: release notes by coderabbit.ai -->

Signed-off-by: Jinzhe Zeng <[email protected]>

backend/find_tensorflow.py

@@ -156,18 +156,15 @@ def get_tf_requirement(tf_version: str = "") -> dict:
                 "tensorflow; platform_machine=='aarch64' or (platform_machine=='arm64' and platform_system == 'Darwin')",
                 # https://github.com/tensorflow/tensorflow/issues/61830
                 "tensorflow-cpu!=2.15.*; platform_system=='Windows'",
-                # TODO: build(wheel): unpin h5py on aarch64
-                # Revert after https://github.com/h5py/h5py/issues/2408 is fixed;
-                # or set UV_PREFER_BINARY when https://github.com/astral-sh/uv/issues/1794 is resolved.
-                # 3.6.0 is the first version to have aarch64 wheels.
-                "h5py>=3.6.0,<3.11.0; platform_system=='Linux' and platform_machine=='aarch64'",
+                # https://github.com/h5py/h5py/issues/2408
+                "h5py>=3.6.0,!=3.11.0; platform_system=='Linux' and platform_machine=='aarch64'",
                 *extra_requires,
             ],
             "gpu": [
                 "tensorflow",
                 "tensorflow-metal; platform_machine=='arm64' and platform_system == 'Darwin'",
                 # See above.
-                "h5py>=3.6.0,<3.11.0; platform_system=='Linux' and platform_machine=='aarch64'",
+                "h5py>=3.6.0,!=3.11.0; platform_system=='Linux' and platform_machine=='aarch64'",
                 *extra_requires,
             ],
             **extra_select,

pyproject.toml

@@ -44,7 +44,7 @@ dependencies = [
     'typing_extensions; python_version < "3.8"',
     'importlib_metadata>=1.4; python_version < "3.8"',
     'h5py',
-    "h5py>=3.6.0,<3.11.0; platform_system=='Linux' and platform_machine=='aarch64'",
+    "h5py>=3.6.0,!=3.11.0; platform_system=='Linux' and platform_machine=='aarch64'",
     'wcmatch',
     'packaging',
     'ml_dtypes',