Skip to content

Commit

Permalink
SEARCH_ALG_IMPORT from ray.search instead of ray.suggest for ray >= 1…
Browse files Browse the repository at this point in the history
….14 (#2196)
  • Loading branch information
jppgks authored Jun 27, 2022
1 parent d404a67 commit edc7c26
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion ludwig/hyperopt/execution.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,13 +37,15 @@
from ray import tune
from ray.tune import register_trainable, Stopper
from ray.tune.schedulers.resource_changing_scheduler import DistributeResources, ResourceChangingScheduler
from ray.tune.suggest import BasicVariantGenerator, ConcurrencyLimiter, SEARCH_ALG_IMPORT
from ray.tune.suggest import BasicVariantGenerator, ConcurrencyLimiter

_ray_114 = version.parse(ray.__version__) >= version.parse("1.14")
if _ray_114:
from ray.tune.search import SEARCH_ALG_IMPORT
from ray.tune.syncer import get_node_to_storage_syncer, SyncConfig
else:
from ray.tune.syncer import get_cloud_sync_client
from ray.tune.suggest import SEARCH_ALG_IMPORT

from ray.tune.utils import wait_for_gpu
from ray.tune.utils.placement_groups import PlacementGroupFactory
Expand Down

0 comments on commit edc7c26

Please sign in to comment.