Commit 8f8f791c

oscar-king <oscar-king@users.noreply.github.com>
2022-06-28 01:14:39
Update setup.py (#99)
* Update setup.py Add missing dependency scikit-learn * Update setup.py Versioning issue fix * Added requirements into extras_require
main
1 parent 53e5ba4
Changed files (1)
setup.py
setup.py
@@ -18,9 +18,15 @@ setup(
         "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": [