具有多个返回并调用ray.get()的ray.get示例

Question

下面的代码执行所需的行为。是否有可能传递前两个函数的第二个参数而不必过早调用ray.get？

@ray.remote
def color():
    image=cv2.imread("frame30.png", flags=0)
    argument= "Hello"
    return image,argument

@ray.remote
def black():
    image=cv2.imread("frame30.png", flags=0)
    argument= "world"
    return image,argument

@ray.remote
def concate_two_args(a,b):
    return a + " " + b

col= color.remote()
blk= black.remote()

#Do I have to "ray.get" in order to pass the results to concate_two_args?
temp1= ray.get(col)[1]
temp2= ray.get(blk)[1]

results= concate_two_args.remote(temp1,temp2)

ray.get(results)

直接这样做

col, string= color.remote()

ray.get(string)

返回

TypeError: cannot unpack non-iterable ray._raylet.ObjectRef object

Answer 1

您能尝试将num_returns添加到@ray.remote吗？例如,

@ray.remote(num_returns=2)
def color():
    image=cv2.imread("frame30.png", flags=0)
    argument= "Hello"
    return image,argument

@ray.remote(num_returns=2)
def black():
    image=cv2.imread("frame30.png", flags=0)
    argument= "world"
    return image,argument