From cd7af48890ba07e556d1b24bf65d5963b4237ef6 Mon Sep 17 00:00:00 2001 From: Avi Shinnar Date: Fri, 4 Apr 2025 09:46:54 -0400 Subject: [PATCH] make tensorflow dependency conditional on newer versions of python Signed-off-by: Avi Shinnar --- .github/workflows/build.yml | 6 +++--- setup.py | 2 +- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build.yml b/.github/workflows/build.yml index 07b172d5f..34b72ea6b 100644 --- a/.github/workflows/build.yml +++ b/.github/workflows/build.yml @@ -155,7 +155,7 @@ jobs: python-version: 3.9 setup-target: '.[full,test]' - test-case: test/test_aif360.py - python-version: 3.9 + python-version: '3.11' setup-target: '.[full,test]' - test-case: test/test_relational_sklearn.py python-version: '3.11' @@ -239,7 +239,7 @@ jobs: python-version: 3.9 setup-target: '.[full,test]' - test-case: test/test_aif360.py - python-version: 3.9 + python-version: '3.11' setup-target: '.[full,test]' - test-case: test/test_relational_sklearn.py python-version: '3.11' @@ -502,7 +502,7 @@ jobs: test-case: - test/test_aif360.py - test/test_aif360_ensembles.py - python-version: [3.9] + python-version: ['3.11'] setup-target: ['.[fairness]'] steps: - uses: actions/checkout@v4 diff --git a/setup.py b/setup.py index 01f0077c8..543c5789b 100644 --- a/setup.py +++ b/setup.py @@ -70,7 +70,7 @@ "lightgbm<4.7.0", "snapml>=1.7.0rc3,<1.17.0", "liac-arff>=2.4.0", - "tensorflow>=2.4.0,<=2.17.0", + "tensorflow>=2.4.0 ; python_version>='3.10'", "smac<=0.10.0", "numba", "aif360>=0.5.0",