aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--setup.cfg6
1 files changed, 5 insertions, 1 deletions
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