diff --git a/lightgbm_ray/tune.py b/lightgbm_ray/tune.py index cd335d1..007313a 100644 --- a/lightgbm_ray/tune.py +++ b/lightgbm_ray/tune.py @@ -55,7 +55,7 @@ def __new__(cls: type, *args, **kwargs): # TODO(justinvyu): [code_removal] Remove in Ray 2.11. raise DeprecationWarning( "`TuneReportCallback` is deprecated. " - "Use `xgboost_ray.tune.TuneReportCheckpointCallback` instead." + "Use `ray.tune.integration.lightgbm.TuneReportCheckpointCallback` instead." )