pin tf to <2.15 on win due to tensorflow/tensorflow#61830

njzjz · Nov 14, 2023 · 9961967 · 9961967
1 parent 153d26e
commit 9961967
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/backend/find_tensorflow.py b/backend/find_tensorflow.py
@@ -136,6 +136,8 @@ def get_tf_requirement(tf_version: str = "") -> dict:
             "cpu": [
                 "tensorflow-cpu; platform_machine!='aarch64' and (platform_machine!='arm64' or platform_system != 'Darwin')",
                 "tensorflow; platform_machine=='aarch64' or (platform_machine=='arm64' and platform_system == 'Darwin')",
+                # https://github.com/tensorflow/tensorflow/issues/61830
+                "tensorflow-cpu<2.15; platform_system=='Windows'",
                 *extra_requires,
             ],
             "gpu": [