可导入的`多处理.Pool`函数

问题描述

这可能很简单，我只是找不到合适的问题。

如果我使用multiprocessing.Pool执行独立脚本，我知道我应该这样做：

def foo(x):
    return x**2

if __name__ == '__main__':
    with Pool(n_jobs) as p:
        p.map(foo, list_of_inputs)

但如果我想使其成为可导入函数，我假设__name__将不再是'__main__'。这样做安全吗：

def __foo(x):
    return x**2

def bar(list_of_inputs, n_jobs):
    with Pool(n_jobs) as p:
        out = p.map(__foo, list_of_inputs)
    return out

###################################################

from test.py import bar

baz = bar(mylist, 4)

我想是这样的，因为bar只会在调用进程中调用，然后只会在子进程中调用foo，但我只是想知道这通常是否是一种安全的方法。

---

解决方案

我看不出您的代码有任何问题，这些代码在我的python3和python2中运行得很好(如果我不对池使用with语句)。在我看来，这是一种安全的使用multiprocessing的方式，我不明白为什么您会经历这种无限挂。您使用的是什么版本的Python？