update dependencies

ddangelov · Nov 10, 2020 · e4615a2 · e4615a2
1 parent 23ab5cb
commit e4615a2
Showing 1 changed file with 12 additions and 6 deletions.
diff --git a/setup.py b/setup.py
@@ -33,11 +33,17 @@
         'umap-learn',
         'hdbscan',
         'wordcloud',
-        'tensorflow',
-        'tensorflow_hub',
-        'tensorflow_text',
-        'torch',
-        'sentence_transformers',
-      ],
+    ],
+    extras_require={
+        sentence_encoders=[
+            'tensorflow',
+            'tensorflow_hub',
+            'tensorflow_text',
+        ],
+        sentence_transformers=[
+            'torch',
+            'sentence_transformers',
+        ],
+    },
     python_requires='>=3.6',
 )