[tune] Remove deprecated mlflow+wandb aliases (#36860)

Signed-off-by: Kai Fricke <kai@anyscale.com>
Signed-off-by: Justin Yu <justinvyu@anyscale.com>
Co-authored-by: Justin Yu <justinvyu@anyscale.com>

python/ray/air/tests/test_integration_mlflow.py

@@ -9,12 +9,7 @@
 
 from ray._private.dict import flatten_dict
 from ray.train._internal.session import init_session
-from ray.tune.trainable import wrap_function
 from ray.tune.trainable.session import _shutdown as tune_session_shutdown
-from ray.tune.integration.mlflow import (
-    MLflowTrainableMixin,
-    mlflow_mixin,
-)
 from ray.air.integrations.mlflow import MLflowLoggerCallback, setup_mlflow, _NoopModule
 from ray.air._internal.mlflow import _MLflowLoggerUtil
 
@@ -201,24 +196,6 @@ def testMlFlowLoggerLogging(self):
             {"dir": "artifact", "run_id": run.info.run_id},
         )
 
-    def testMlFlowMixinConfig(self):
-        clear_env_vars()
-        trial_config = {"par1": 4, "par2": 9.0}
-
-        with self.assertRaises(DeprecationWarning):
-
-            @mlflow_mixin
-            def train_fn(config):
-                return 1
-
-        def train_fn_2(config):
-            return 1
-
-        train_fn_2.__mixins__ = (MLflowTrainableMixin,)
-
-        with self.assertRaises(DeprecationWarning):
-            wrap_function(train_fn_2)(trial_config)
-
     def testMlFlowSetupExplicit(self):
         clear_env_vars()
         trial_config = {"par1": 4, "par2": 9.0}

python/ray/air/tests/test_integration_wandb.py

@@ -47,9 +47,6 @@
 
 import ray
 from ray.exceptions import RayActorError
-from ray.tune import Trainable
-from ray.tune.integration.wandb import WandbTrainableMixin
-from ray.tune.integration.wandb import wandb_mixin
 from ray.air.integrations.wandb import (
     WandbLoggerCallback,
     _QueueItem,
@@ -453,25 +450,6 @@ def finish(self):
             ray.get(actor.get_state.remote())
 
 
-class TestWandbClassMixin:
-    def test_wandb_mixin(self):
-        class WandbTestTrainable(WandbTrainableMixin, Trainable):
-            pass
-
-        config = {}
-        with pytest.raises(DeprecationWarning):
-            WandbTestTrainable(config)
-
-
-class TestWandbMixinDecorator:
-    def test_wandb_decorator(self):
-        with pytest.raises(DeprecationWarning):
-
-            @wandb_mixin
-            def train_fn(config):
-                return 1
-
-
 def test_wandb_logging_process_run_info_hook(monkeypatch):
     """
     Test WANDB_PROCESS_RUN_INFO_HOOK in _WandbLoggingActor is

python/requirements/ml/tune-requirements.txt

@@ -17,4 +17,4 @@ hyperopt==0.2.7
 nevergrad==0.4.3.post7
 optuna==3.2.0
 
-tune-sklearn==0.4.4
+tune-sklearn==0.4.6