restrict tensorflow version

alteryx · Mar 11, 2024 · 3cb030d · 3cb030d
1 parent 3101721
commit 3cb030d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -64,9 +64,9 @@ dev = [
     "nlp_primitives[test]"
 ]
 complete = [
-    "tensorflow >= 1.14.0; sys_platform!='darwin' or platform_machine!='arm64'",
+    "tensorflow >= 1.14.0, < 2.16.0; sys_platform!='darwin' or platform_machine!='arm64'",
     "tensorflow-metal >= 0.4.0; sys_platform=='darwin' and platform_machine=='arm64'",
-    "tensorflow-macos >= 2.8.0; sys_platform=='darwin' and platform_machine=='arm64'",
+    "tensorflow-macos >= 2.8.0, < 2.16.0; sys_platform=='darwin' and platform_machine=='arm64'",
     "tensorflow_hub >= 0.4.0, < 0.16.0",
 ]