[Cherry-pick] Downgrade scikit-learn for python3.6 (#1042) (#1043)

* Downgrade scikit-learn for python3.6 (#1042)

* Fix paths for training_args.py (#1029)

* Fix paths for training_args.py

* Fix quality

Co-authored-by: Rahul Tuli <rahul@neuralmagic.com>

Co-authored-by: Eldar Kurtic <eldar.ciki@gmail.com>
Co-authored-by: Rahul Tuli <rahul@neuralmagic.com>

Author: 3 people

setup.py

@@ -46,7 +46,7 @@
     "pydantic>=1.5.0",
     "requests>=2.0.0",
     "scikit-image>=0.15.0",
-    "scikit-learn>=1.0.2",
+    "scikit-learn>=0.24.2",
     "scipy>=1.0.0",
     "tqdm>=4.0.0",
     "toposort>=1.0",

src/sparseml/transformers/masked_language_modeling.py

@@ -289,7 +289,8 @@ def __post_init__(self):
 
 @record
 def main(**kwargs):
-    # See all possible arguments in src/transformers/training_args.py
+    # See all possible arguments in
+    # src/sparseml/transformers/sparsification/training_args.py
     # or by passing the --help flag to this script.
     # We now keep distinct sets of args, for a cleaner separation of concerns.
 

src/sparseml/transformers/question_answering.py

@@ -307,7 +307,8 @@ def __post_init__(self):
 
 @record
 def main(**kwargs):
-    # See all possible arguments in src/transformers/training_args.py
+    # See all possible arguments in
+    # src/sparseml/transformers/sparsification/training_args.py
     # or by passing the --help flag to this script.
     # We now keep distinct sets of args, for a cleaner separation of concerns.
 

src/sparseml/transformers/text_classification.py

@@ -296,7 +296,8 @@ class ModelArguments:
 
 @record
 def main(**kwargs):
-    # See all possible arguments in src/transformers/training_args.py
+    # See all possible arguments in
+    # src/sparseml/transformers/sparsification/training_args.py
     # or by passing the --help flag to this script.
     # We now keep distinct sets of args, for a cleaner separation of concerns.
 

src/sparseml/transformers/token_classification.py

@@ -260,7 +260,8 @@ def __post_init__(self):
 
 @record
 def main(**kwargs):
-    # See all possible arguments in src/transformers/training_args.py
+    # See all possible arguments in
+    # src/sparseml/transformers/sparsification/training_args.py
     # or by passing the --help flag to this script.
     # We now keep distinct sets of args, for a cleaner separation of concerns.