diff --git a/setup.py b/setup.py
index ebc6dba2..84ef8937 100644
--- a/setup.py
+++ b/setup.py
@@ -40,6 +40,7 @@
     "black",
     "pytest",
     "coverage",
+    "scikit-learn",
 ]
 
 setuptools.setup(