Use scikit-learn instead of sklearn (#1294)

neuralmagic · Mar 13, 2023 · d00b6d9 · d00b6d9
1 parent 9647dae
commit d00b6d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",
             ]
         )