如何导入具有MPI功能的模块并在我的模块中使用它？

Question

我正在制作一个依赖于另一个具有MPI功能的模块的模块。当使用MPI模块作为开发人员时，我应该......

# script_one.py
from mpi_module import *
some_mpi_module_function()

...并使用mpirun -n 4 script_one.py执行一切正常工作，复杂的任务在some_mpi_module_function()内的处理器之间分配。

在我的模块中，我反对开发人员在没有namspaces的情况下导入的建议：

# my_module
import mpi_module
def my_module_function():
    some_task()
    mpi_module.some_mpi_module_function()

但是，当我从脚本中导入模块时......

# script_two.py
import my_module
my_module.my_module_function()

...我运行mpirun -n 4 script_two.py，我的脚本运行了四个实例，每个处理器依次运行mpi_module.some_mpi_module_function()。

如何让我的模块使用其他模块的MPI功能？