Python Arg 예제들

예제 #1

파일: util.py 프로젝트: yukingx/ray

def load_args_proto(thing):
    arg = ray_client_pb2.Arg()
    if isinstance(thing, ClientObjectRef):
        arg.local = ray_client_pb2.Arg.Locality.REFERENCE
        arg.reference_id = thing.id
    else:
        arg.local = ray_client_pb2.Arg.Locality.INTERNED
        arg.data = cloudpickle.dumps(thing)
    return arg

예제 #2

def convert_to_arg(val):
    out = ray_client_pb2.Arg()
    if isinstance(val, ClientObjectRef):
        out.local = ray_client_pb2.Arg.Locality.REFERENCE
        out.reference_id = val.id
    else:
        out.local = ray_client_pb2.Arg.Locality.INTERNED
        out.data = cloudpickle.dumps(val)
    return out

예제 #3

파일: client_pickler.py 프로젝트: holdenk/ray

def convert_to_arg(val: Any, client_id: str) -> ray_client_pb2.Arg:
    out = ray_client_pb2.Arg()
    out.local = ray_client_pb2.Arg.Locality.INTERNED
    out.data = dumps_from_client(val, client_id)
    return out