Update setup.py (#99)

* Update setup.py

Add missing dependency scikit-learn

* Update setup.py

Versioning issue fix

* Added requirements into extras_require

Author: oscar-king

setup.py

@@ -18,9 +18,15 @@
         "tqdm",  # Needed for progress bars
         "pandas>=1.2.3",  # Needed for CLI fine-tuning data preparation tool
         "pandas-stubs>=1.1.0.11",  # Needed for type hints for mypy
-        "openpyxl>=3.0.7",  # Needed for CLI fine-tuning data preparation tool xlsx format
+        "openpyxl>=3.0.7"  # Needed for CLI fine-tuning data preparation tool xlsx format
     ],
-    extras_require={"dev": ["black~=21.6b0", "pytest==6.*"]},
+    extras_require={
+        "dev": ["black~=21.6b0", "pytest==6.*"],
+        "embeddings": [
+            "scikit-learn>=1.0.2", # Needed for embedding utils, versions >= 1.1 require python 3.8
+            "tenacity>=8.0.1"
+        ],
+    },
     python_requires=">=3.7.1",
     entry_points={
         "console_scripts": [