From df9035360bfbbb3e6ae528dbe25f290cf44ce4db Mon Sep 17 00:00:00 2001
From: Toby Roseman <troseman@apple.com>
Date: Wed, 29 Jul 2020 14:08:44 -0700
Subject: [PATCH] Relax TensorFlow Requirement

---
 src/python/setup.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/python/setup.py b/src/python/setup.py
index 8d0a115885..587c2d643b 100644
--- a/src/python/setup.py
+++ b/src/python/setup.py
@@ -182,9 +182,9 @@ def do_not_install(require):
     if sys.platform == "darwin":
         install_requires.append("tensorflow >= 2.0.0")
     else:
-        # ST, OD, AC and DC segfault on Linux with TensorFlow 2.1
+        # ST, OD, AC and DC segfault on Linux with TensorFlow 2.1.0 and 2.1.1
         # See: https://github.com/apple/turicreate/issues/3003
-        install_requires.append("tensorflow >= 2.0.0,<= 2.0.1 ")
+        install_requires.append("tensorflow >= 2.0.0,!= 2.1.0,!= 2.1.1")
 
     setup(
         name="turicreate",