mirror of
https://github.com/vale981/ray
synced 2025-03-08 19:41:38 -05:00
e0867c8845
* 4 space indentation for actor.py. * 4 space indentation for worker.py. * 4 space indentation for more files. * 4 space indentation for some test files. * Check indentation in Travis. * 4 space indentation for some rl files. * Fix failure test. * Fix multi_node_test. * 4 space indentation for more files. * 4 space indentation for remaining files. * Fixes.
35 lines
998 B
Python
35 lines
998 B
Python
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import os
|
|
import time
|
|
|
|
import ray
|
|
|
|
|
|
@ray.remote
|
|
def f():
|
|
time.sleep(0.1)
|
|
return ray.services.get_node_ip_address()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
driver_index = int(os.environ["RAY_DRIVER_INDEX"])
|
|
redis_address = os.environ["RAY_REDIS_ADDRESS"]
|
|
print("Driver {} started at {}.".format(driver_index, time.time()))
|
|
|
|
ray.init(redis_address=redis_address)
|
|
# Check that tasks are scheduled on all nodes.
|
|
num_attempts = 30
|
|
for i in range(num_attempts):
|
|
ip_addresses = ray.get([f.remote() for i in range(1000)])
|
|
distinct_addresses = set(ip_addresses)
|
|
counts = [ip_addresses.count(address) for address
|
|
in distinct_addresses]
|
|
print("Counts are {}".format(counts))
|
|
if len(counts) == 5:
|
|
break
|
|
assert len(counts) == 5
|
|
|
|
print("Driver {} finished at {}.".format(driver_index, time.time()))
|