From 61c8b8f9d330c199230b9764131458ab69306b52 Mon Sep 17 00:00:00 2001 From: Agustin Picard Date: Tue, 27 Aug 2024 16:17:10 +0200 Subject: [PATCH] ci: modify the setup to limit tf to 2.15 as 2.16 breaks a lot of things --- setup.cfg | 14 +++++++------- setup.py | 2 +- 2 files changed, 8 insertions(+), 8 deletions(-) diff --git a/setup.cfg b/setup.cfg index 3fde7c5a..8430629c 100644 --- a/setup.cfg +++ b/setup.cfg @@ -41,10 +41,10 @@ commands = deps = pytest pytest-cov - tf21: tensorflow ~= 2.2.0 - tf25: tensorflow ~= 2.5.0 - tf28: tensorflow ~= 2.8.0 - tf211: tensorflow ~= 2.11.0 + tf21: tensorflow ~= 2.2.0,<2.16 + tf25: tensorflow ~= 2.5.0,<2.16 + tf28: tensorflow ~= 2.8.0,<2.16 + tf211: tensorflow ~= 2.11.0,<2.16 -rrequirements.txt commands = pytest --cov=xplique --ignore=xplique/wrappers/pytorch.py --ignore=tests/wrappers/test_pytorch_wrapper.py --ignore=tests/concepts/test_craft_torch.py {posargs} @@ -53,9 +53,9 @@ commands = deps = pytest pytest-cov - tf25: tensorflow ~= 2.5.0 - tf28: tensorflow ~= 2.8.0 - tf211: tensorflow ~= 2.11.0 + tf25: tensorflow ~= 2.5.0,<2.16 + tf28: tensorflow ~= 2.8.0,<2.16 + tf211: tensorflow ~= 2.11.0,<2.16 torch111: torch == 1.11.0 torch113: torch == 1.13.0 torch200: torch diff --git a/setup.py b/setup.py index b5dc14d7..04c2f989 100644 --- a/setup.py +++ b/setup.py @@ -12,7 +12,7 @@ author="Thomas FEL", author_email="thomas_fel@brown.edu", license="MIT", - install_requires=['tensorflow>=2.1.0', 'numpy', 'scikit-learn', 'scikit-image', + install_requires=['tensorflow>=2.1.0,<2.16.0', 'numpy', 'scikit-learn', 'scikit-image', 'matplotlib', 'scipy', 'opencv-python', 'deprecated'], extras_require={ "tests": ["pytest", "pylint"],