bumping allowed version of tensorflow

py4dstem · Dec 18, 2023 · 870eb00 · 870eb00
1 parent 11b64a9
commit 870eb00
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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",