diff options
author | Gergely Nagy <gergely.nagy@arm.com> | 2024-04-11 17:34:18 +0100 |
---|---|---|
committer | Gergely Nagy <gergely.nagy@arm.com> | 2024-04-12 09:31:20 +0100 |
commit | 4de782fde8e38ec92bb5bc60e156de027f13bfba (patch) | |
tree | 65e69f4b95b58214fc27261575dfc80a40023705 | |
parent | 92452bda7dd8268fb9eb6022090187d779914bbf (diff) | |
download | mlia-4de782fde8e38ec92bb5bc60e156de027f13bfba.tar.gz |
fix: Pin h5py + TF dependency to fix issue on aarch64
Worked around dependency issue on aarch64, due to this
bug: https://github.com/h5py/h5py/issues/2408.
When the bug is fixed, this commit can be reverted.
Pinning the following dependencies:
* tensorflow==2.15.1
* h5py==3.10.0
Change-Id: Iada706c403f14d29735e9abea986427bbdae0660
Signed-off-by: Gergely Nagy <gergely.nagy@arm.com>
-rw-r--r-- | setup.cfg | 6 |
1 files changed, 5 insertions, 1 deletions
@@ -28,8 +28,12 @@ python_requires = >=3.9.0 package_dir = = src packages = find_namespace: +# Pinning tensorflow & h5py to work around build issue on aarch64: +# https://github.com/h5py/h5py/issues/2408 +# Idea is to unpin these when it's resolved. install_requires = - tensorflow~=2.15.1 + tensorflow==2.15.1 + h5py==3.10.0 tensorflow-model-optimization~=0.7.5 ethos-u-vela~=3.11.0 flaky~=3.7.0 |