diff --git a/pyproject.toml b/pyproject.toml
index 2ab07d1..deeedf4 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -28,7 +28,8 @@ dependencies = [
     "pydantic",
     "sympy",
     "scikit-rf",
-    "setuptools"
+    "setuptools",
+    "scikit-learn"
 ]
 
 [project.urls]