mirror of
https://github.com/vale981/ray
synced 2025-03-09 12:56:46 -04:00
40 lines
915 B
Python
40 lines
915 B
Python
|
from __future__ import absolute_import
|
||
|
|
from __future__ import division
|
||
|
|
from __future__ import print_function
|
||
|
|
|
||
|
|
import time
|
||
|
|
|
||
|
|
import ray
|
||
|
|
|
||
|
|
|
||
|
|
def setup(*args):
|
||
|
|
if not hasattr(setup, "is_initialized"):
|
||
|
|
ray.init(num_workers=4, num_cpus=4)
|
||
|
|
setup.is_initialized = True
|
||
|
|
|
||
|
|
|
||
|
|
@ray.remote
|
||
|
|
def sleep(x):
|
||
|
|
time.sleep(x)
|
||
|
|
|
||
|
|
|
||
|
|
class WaitSuite(object):
|
||
|
|
timeout = 10
|
||
|
|
timer = time.time
|
||
|
|
|
||
|
|
def time_wait_task(self):
|
||
|
|
ray.wait([sleep.remote(0.1)])
|
||
|
|
|
||
|
|
def time_wait_many_tasks(self, num_returns):
|
||
|
|
tasks = [sleep.remote(i / 5) for i in range(4)]
|
||
|
|
ray.wait(tasks, num_returns=num_returns)
|
||
|
|
|
||
|
|
time_wait_many_tasks.params = list(range(1, 4))
|
||
|
|
time_wait_many_tasks.param_names = ["num_returns"]
|
||
|
|
|
||
|
|
def time_wait_timeout(self, timeout):
|
||
|
|
ray.wait([sleep.remote(0.5)], timeout=timeout)
|
||
|
|
|
||
|
|
time_wait_timeout.params = [200, 800]
|
||
|
|
time_wait_timeout.param_names = ["timeout_ms"]
|