diff options
Diffstat (limited to 'setup.cfg')
-rw-r--r-- | setup.cfg | 8 |
1 files changed, 6 insertions, 2 deletions
@@ -1,4 +1,4 @@ -# SPDX-FileCopyrightText: Copyright 2022-2023, Arm Limited and/or its affiliates. +# SPDX-FileCopyrightText: Copyright 2022-2024, Arm Limited and/or its affiliates. # SPDX-FileCopyrightText: Copyright (c) 2020 Troy Comi # SPDX-License-Identifier: Apache-2.0 AND MIT @@ -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 |