Question

我有2个文件：project / pipline.py和project / partA / algorithm.py。

在algorithm.py文件中，我具有主要功能：

if __name__ == '__main__':
     parser = argparse.ArgumentParser()
     parser.add_argument('--a', type=str)
     parser.add_argument('--b', type=str)
     parser.add_argument('--c', type=str)

我想通过pipline脚本调用此函数，

我尝试过：

from partA.algorithm import main
main()

但是它不起作用。我该怎么办？如何发送论点？

Answer 1

您需要在algorithm.py文件中添加它：

def main():
    parser = argparse.ArgumentParser()
    parser.add_argument('--a', type=str)
    parser.add_argument('--b', type=str)
    parser.add_argument('--c', type=str)

if __name__ == '__main__':
    main()

第if __name__ == '__main__':行仅检查python文件是否未作为导入模块运行，如果是，则在代码块中运行代码。它没有定义可以导入的功能。

如何从另一个文件导入main

1 个答案: