From 840dd368be3ee9cb2e535bb7adc761e1daad0c58 Mon Sep 17 00:00:00 2001
From: Tomasz Latkowski <tlatkowski@gmail.com>
Date: Mon, 6 Jul 2020 08:25:09 +0200
Subject: [PATCH] added extra req

---
 setup.py | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/setup.py b/setup.py
index d694061..25414f8 100644
--- a/setup.py
+++ b/setup.py
@@ -1,5 +1,4 @@
 from setuptools import setup
-from setuptools import Extension
 
 with open('requirements/requirements.txt') as f:
     requirements = f.readlines()
@@ -11,9 +10,12 @@
     description='GANs2.0',
     long_description=(
         'Generative Adversarial Networks in TensorFlow 2.0.'
-        ),
+    ),
     long_description_content_type='text/markdown',
     install_requires=requirements,
+    extras_require={
+        'tensorflow': ['tensorflow>=2.2.0'],
+    },
     packages=[
         'gans.datasets',
         'gans.layers',
@@ -22,4 +24,4 @@
     ],
     zip_safe=True
 )
-'Generative Adversarial Networks in TensorFlow 2.0'
\ No newline at end of file
+'Generative Adversarial Networks in TensorFlow 2.0'