diff --git a/common/src/autogluon/common/utils/try_import.py b/common/src/autogluon/common/utils/try_import.py index 16ae951a8a8d..25806c61c56a 100644 --- a/common/src/autogluon/common/utils/try_import.py +++ b/common/src/autogluon/common/utils/try_import.py @@ -32,7 +32,7 @@ def try_import_mxboard(): def try_import_ray() -> ModuleType: - RAY_MAX_VERSION = "2.49.0" # sync with core/setup.py + RAY_MAX_VERSION = "2.50.0" # sync with core/setup.py ray_max_version_os_map = dict( Darwin=RAY_MAX_VERSION, Windows=RAY_MAX_VERSION, diff --git a/core/setup.py b/core/setup.py index 93ab3514543a..d741ef210f3d 100644 --- a/core/setup.py +++ b/core/setup.py @@ -48,11 +48,11 @@ extras_require = { "ray": [ - "ray[default]>=2.10.0,<2.49", # sync with common/src/autogluon/common/utils/try_import.py + "ray[default]>=2.10.0,<2.50", # sync with common/src/autogluon/common/utils/try_import.py ], "raytune": [ "pyarrow>=15.0.0", # cap Pyarrow to fix source installation - https://github.com/autogluon/autogluon/issues/4519 - "ray[default,tune]>=2.10.0,<2.49", # sync with common/src/autogluon/common/utils/try_import.py + "ray[default,tune]>=2.10.0,<2.50", # sync with common/src/autogluon/common/utils/try_import.py # TODO: consider alternatives as hyperopt is not actively maintained. "hyperopt>=0.2.7,<0.2.8", # This is needed for the bayes search to work. # 'GPy>=1.10.0,<1.11.0' # TODO: Enable this once PBT/PB2 are supported by ray lightning