From 5035a76a8e1235a2a0a45c2b4107ba0424d6c083 Mon Sep 17 00:00:00 2001
From: corey-nm <109536191+corey-nm@users.noreply.github.com>
Date: Thu, 5 Jan 2023 12:06:17 -0500
Subject: [PATCH] Use scikit-learn instead of sklearn

---
 src/sparseml/transformers/__init__.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sparseml/transformers/__init__.py b/src/sparseml/transformers/__init__.py
index 891e5cd5b44..36aadd69d85 100644
--- a/src/sparseml/transformers/__init__.py
+++ b/src/sparseml/transformers/__init__.py
@@ -56,7 +56,7 @@ def _install_transformers_and_deps():
                 "install",
                 transformers_requirement,
                 "datasets<=1.18.4",
-                "sklearn",
+                "scikit-learn",
                 "seqeval",
             ]
         )