ray.services.get_node_ip_address -> ray.util.get_node_ip_address (#154)

Author: edoakes

xgboost_ray/main.py

@@ -24,10 +24,9 @@
 try:
     import ray
     from ray import logger
-    from ray.services import get_node_ip_address
     from ray.exceptions import RayActorError, RayTaskError
     from ray.actor import ActorHandle
-    from ray.util import placement_group
+    from ray.util import get_node_ip_address, placement_group
     from ray.util.placement_group import PlacementGroup, \
         remove_placement_group, get_current_placement_group
 

xgboost_ray/tests/release/tune_placement.py

@@ -30,9 +30,9 @@
 import ray
 
 from ray import tune
-from ray.services import get_node_ip_address
 from ray.tune.session import get_trial_id
 from ray.tune.integration.docker import DockerSyncer
+from ray.util import get_node_ip_address
 
 from benchmark_cpu_gpu import train_ray
 from xgboost_ray import RayParams