diff --git a/requirements.txt b/requirements.txt
index cc3aafd..7542d1b 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -6,7 +6,7 @@ mpmath
 numpy~=1.21.5
 pandas~=1.1.4
 scipy~=1.2.1
-sklearn~=1.0.2
+scikit-learn~=1.0.2
 tensorflow-datasets~=4.5.2
 tensorflow-estimator~=2.4
 tensorflow-probability~=0.15.0
diff --git a/setup.py b/setup.py
index 842c141..5a2101c 100644
--- a/setup.py
+++ b/setup.py
@@ -34,7 +34,7 @@ setup(
         'numpy~=1.21.5',
         'pandas~=1.1.4',
         'scipy~=1.2.1',
-        'sklearn~=1.0.2',
+        'scikit-learn~=1.0.2',
         'tensorflow-datasets~=4.5.2',
         'tensorflow-estimator~=2.4',
         'tensorflow-probability~=0.15.0',