diff --git a/pyproject.toml b/pyproject.toml
index ae354a5..d819931 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -23,6 +23,7 @@ dependencies = [
     "gseapy",
     "torch",
     "scipy",
+    "scikit-learn",
     "tqdm",
     "matplotlib",
     "numpy<2",
@@ -34,4 +35,4 @@ requires-python = ">=3.8"
 
 [project.urls]
 Home = "https://teichlab.github.io/Genes2Genes"
-Repository = "https://github.com/Teichlab/Genes2Genes"
\ No newline at end of file
+Repository = "https://github.com/Teichlab/Genes2Genes"