从另一个 Python 脚本中启动 IPython Parallel

Question

假设我有两个 Python 文件

test.py

from ipyparallel import Client

def hi(a):
    return b + (a * 2)

def run():
    b = 3

    client = Client()
    view = client[:]

    view.push({'b':b})
    results = view.map(hi, [0,1,2,3,4])
    for r in results:
        print(r)

和driver.py

from test import run

if __name__ == '__main__':
    run()

我收到错误 [0:apply]: NameError: name 'b' is not defined。

如果我从 test.py 中调用 run()，此代码将起作用，但是，我不想这样做。我想从 driver.py 中调用 run()。有关如何解决此问题的任何想法？

Answer 1

在test.py 文件中导入interactive，然后在map 函数中使用interactive(hi) 而不是hi：

from ipyparallel import Client
from ipyparallel.util import interactive

def hi(a):
    global b
    return b + (a * 2)

def run():
    b = 3
    client = Client()
    view = client[:]
    view.push({'b':b})
    results = view.map(interactive(hi), [0,1,2,3,4])
    for r in results:
        print(r)

if __name__ == '__main__':
    run()