From 177b459b6fedc9f28f42db78cae6bf9bdeef8f8a Mon Sep 17 00:00:00 2001
From: Parisa Mojiri <parisa.mojiri@gmail.com>
Date: Wed, 27 Oct 2021 00:00:03 -0400
Subject: [PATCH] Update setup.py

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

diff --git a/setup.py b/setup.py
index 3bc3de4..f8f086f 100644
--- a/setup.py
+++ b/setup.py
@@ -40,8 +40,8 @@
         'keras-contrib', 'pandas', 'numpy', 'plotly', 'PyQt5', 'termcolor'
     ],
     extras_require={
-        "icvmapper": ["tensorflow"],
-        "icvmapper_gpu": ["tensorflow-gpu"],
+        "icvmapper": ["tensorflow==1.15"],
+        "icvmapper_gpu": ["tensorflow-gpu==1.15"],
     },
     entry_points={'console_scripts': ['icvmapper=icvmapper.cli:main']},
     keywords=[