[Train] Remove deprecated mosaic and sklearn trainer code (ray-projec…

…t#47901) **Both trainers are now hard-deprecated.** This PR removes the code paths that confuses Ray train users on Mosaic and Sklearn. This should have been done a while ago. Signed-off-by: Superskyyy <yihaochen@apache.org> Signed-off-by: ujjawal-khare <ujjawal.khare@dream11.com>
ujjawal-khare-27 · Oct 15, 2024 · 0befe07 · 0befe07
1 parent 60c8554
commit 0befe07
Show file tree

Hide file tree

Showing 11 changed files with 0 additions and 626 deletions.
diff --git a/python/ray/air/_internal/usage.py b/python/ray/air/_internal/usage.py
@@ -17,12 +17,9 @@
 AIR_TRAINERS = {
     "HorovodTrainer",
     "LightGBMTrainer",
-    "MosaicTrainer",
-    "SklearnTrainer",
     "TensorflowTrainer",
     "TorchTrainer",
     "XGBoostTrainer",
-    "HuggingFaceTrainer",  # Deprecated: Remove in 2.7.
 }
 
 # searchers implemented by Ray Tune.

diff --git a/python/ray/train/BUILD b/python/ray/train/BUILD
@@ -487,14 +487,6 @@ py_test(
     deps = [":train_lib", ":conftest"]
 )
 
-py_test(
-    name = "test_sklearn_predictor",
-    size = "small",
-    srcs = ["tests/test_sklearn_predictor.py"],
-    tags = ["team:ml", "exclusive", "ray_air"],
-    deps = [":train_lib", ":conftest"]
-)
-
 py_test(
     name = "test_storage",
     size = "small",

diff --git a/python/ray/train/examples/mosaic_cifar10_example.py b/python/ray/train/examples/mosaic_cifar10_example.py
diff --git a/python/ray/train/mosaic/__init__.py b/python/ray/train/mosaic/__init__.py
diff --git a/python/ray/train/mosaic/_mosaic_utils.py b/python/ray/train/mosaic/_mosaic_utils.py
diff --git a/python/ray/train/mosaic/mosaic_trainer.py b/python/ray/train/mosaic/mosaic_trainer.py
diff --git a/python/ray/train/sklearn/__init__.py b/python/ray/train/sklearn/__init__.py
diff --git a/python/ray/train/sklearn/sklearn_checkpoint.py b/python/ray/train/sklearn/sklearn_checkpoint.py