diff --git a/setup.cfg b/setup.cfg
index a42b05ef01..3729647b8d 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -45,9 +45,9 @@ embeddings =
   scikit-learn >= 1.0.2  # Needed for embedding utils, versions >= 1.1 require python 3.8
   tenacity >= 8.0.1
   matplotlib
-  sklearn
   plotly
   numpy
+  scipy
   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