From 99fbb9165a5e07f9eb9da706f66ea41454e1b006 Mon Sep 17 00:00:00 2001 From: Justin Yu Date: Sun, 4 Feb 2024 10:35:32 -0800 Subject: [PATCH] Fix deprecation msg Signed-off-by: Justin Yu --- lightgbm_ray/tune.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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." )