python - 如何从相对目录(在 Jupyter 笔记本中)加载自定义单元魔法？

Question

我有一个自定义单元格魔法，我需要从相对路径加载。当模块目录与笔记本位于同一位置时，我可以这样做。但是，当我有更复杂的目录结构时，它不起作用。

如果目录结构是这样的，它就可以工作:

test_custom_magic\
    |-custom_magic_code\
        |-__init__.py
        |-etcetc.py
    |-test_notebook.ipy

在这种情况下，我只需执行 %reload_ext custom_magic_code 并且我的代码就可以工作。

但是，现在我已经完成了代码的编写和测试，我正在尝试在更复杂的目录中使用它:

important_project\
    |-notebooks\
        |-do_something_important.ipy
        |-do_something_else_important.ipy
    |-custom_magic_code\
        |-__init__.py
        |-etcetc.py

在do_something_important.ipy中，我不能只是%reload_ext custom_magic_code。我收到 ModuleNotFoundError。

如果我尝试 %reload_ext ../custom_magic_code (或其他嵌套目录级别)，我会收到此错误:类型错误:需要“package”参数才能执行“../custom_magic_code”的相对导入

如何解决这个问题？请注意，我的自定义魔术代码足以让我在本地运行。我还没有准备好打包它并通过 pip/conda/nbextension 安装(而且我试图避免学习这一点，直到我能让当前的代码正常工作)。

Answer 1

所以我最终使用以下代码在 python 检查的路径中包含相关目录以查找库:

module_path = os.path.abspath(os.path.join('../..'))
if module_path not in sys.path:
    sys.path.append(module_path)

似乎有效!