diff --git a/requirements.txt b/requirements.txt
index 133e211e..0b85ce2d 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,4 +6,4 @@ tiktoken>=0.3.2
 fire
 scipy
 patsy
-sklearn
\ No newline at end of file
+scikit-learn
\ No newline at end of file
diff --git a/setup.py b/setup.py
index 11c92826..7ef6daf6 100644
--- a/setup.py
+++ b/setup.py
@@ -59,7 +59,7 @@
         "scipy",
         "huggingface_hub",
         "patsy",
-        "sklearn",
+        "scikit-learn",
     ],
     extras_require={
         "analysis": PACKAGES_ANALYSIS,