Python _get_runtime_context Examples

Example #1

File: test_advanced_3.py Project: wwxFromTju/ray

def test_specific_job_id():
    dummy_driver_id = ray.JobID.from_int(1)
    ray.init(num_cpus=1, job_id=dummy_driver_id)

    # in driver
    assert dummy_driver_id == ray._get_runtime_context().current_driver_id

    # in worker
    @ray.remote
    def f():
        return ray._get_runtime_context().current_driver_id

    assert dummy_driver_id == ray.get(f.remote())

    ray.shutdown()

Example #2

File: test_advanced_3.py Project: wwxFromTju/ray

 def f():
     return ray._get_runtime_context().current_driver_id