From 4de782fde8e38ec92bb5bc60e156de027f13bfba Mon Sep 17 00:00:00 2001 From: Gergely Nagy Date: Thu, 11 Apr 2024 17:34:18 +0100 Subject: 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 --- setup.cfg | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/setup.cfg b/setup.cfg index 6917747..6ddb576 100644 --- a/setup.cfg +++ b/setup.cfg @@ -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 -- cgit v1.2.1