added extra req

tlatkowski · Jul 6, 2020 · 840dd36 · 840dd36
1 parent 8b4567e
commit 840dd36
Showing 1 changed file with 5 additions and 3 deletions.
diff --git 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'
+'Generative Adversarial Networks in TensorFlow 2.0'