From 870eb00072cc40b1cf3f08154c9822dda8723739 Mon Sep 17 00:00:00 2001 From: alex-rakowski Date: Mon, 18 Dec 2023 01:26:51 -0800 Subject: [PATCH] bumping allowed version of tensorflow --- setup.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/setup.py b/setup.py index c199930f0..58f61dc4a 100644 --- a/setup.py +++ b/setup.py @@ -48,13 +48,13 @@ "cuda": ["cupy >= 10.0.0"], "acom": ["pymatgen >= 2022", "mp-api == 0.24.1"], "aiml": [ - "tensorflow == 2.8.0", + "tensorflow <= 2.10.0", "tensorflow-addons <= 0.16.1", "crystal4D", "typeguard == 2.7", ], "aiml-cuda": [ - "tensorflow == 2.8.0", + "tensorflow <= 2.10.0", "tensorflow-addons <= 0.16.1", "crystal4D", "cupy >= 10.0.0",