mirror of
https://github.com/wassname/ray.git
synced 2026-06-29 00:55:31 +08:00
Simon Mo
22 lines
469 B
Python
22 lines
469 B
Python
"""
|
|
Helper methods for dealing with ray.ObjectID
|
|
"""
|
|
|
|
from __future__ import absolute_import
|
|
from __future__ import division
|
|
from __future__ import print_function
|
|
|
|
import ray
|
|
|
|
|
|
def unwrap(future):
|
|
return ray.get(future)[0]
|
|
|
|
|
|
def get_new_oid():
|
|
worker = ray.worker.global_worker
|
|
oid = ray._raylet.compute_put_id(worker.current_task_id,
|
|
worker.task_context.put_index)
|
|
worker.task_context.put_index += 1
|
|
return oid
|