From 2b1c778d4a35a9362ddfd64fabe29802c5957ea3 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fabian=20H=C3=B6ring?= <f.horing@criteo.com>
Date: Thu, 29 Oct 2020 10:26:54 +0100
Subject: [PATCH] Allow tf < 3.0

- tested with tf 2.3.1
- integration tests are executed with tf 1.15.2 & tf 2.2.0

Change-Id: I6c51c9487f517fd5a9fbc1f8f22886558664ffe8
---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 3cad8db..a89032f 100755
--- a/setup.py
+++ b/setup.py
@@ -42,7 +42,7 @@ def _check_add_criteo_environment(package_name):
 
 
 def get_tensorflow_version() -> str:
-    version = ">=1.15.0,<=2.2.0"
+    version = ">=1.15.0,<3.0"
     if "BUILD_GPU" in os.environ:
         return f"tensorflow_gpu{version}"
     else: