モジュールのインポートが出来ないのですが。。。

Question

こちらのプログラムを実行しようとしています。https://github.com/jakesnell/prototypical-networks
そこで以上のページにある様に、python scripts/train/few_shot/run_train.pyで実行しようとしました。しかしそこで

/prototypical-networks-master/scripts/train/few_shot/train.py", line 16, in <module>
ModuleNotFoundError: No module named 'protonets.utils'
といったエラーが出ます。

from protonets.engine import Engine #ここは問題無い

import protonets.utils.data as data_utils #ここでエラー
import protonets.utils.model as model_utils
import protonets.utils.log as log_utils

以上の最初の1行はインポートできているのになぜその下はダメなのでしょう？ディレクトリの問題の部分は以下の様になっています。

prototypical-networks-master
                            `-- protonets
                                         `-- engine.py, --utils
                                                               `-- data.py,model.py,log,py

prototypical-networks-masterからpython scripts/train/few_shot/run_train.pyを実行しています。

Answer 1

Insrall dependenciesのところに

Install the protonets package by running python setup.py install or python setup.py develop.

と書いてありますが、python setup.py installでは駄目でpython setup.py developの方が正解っぽいですね。
setuptoolsにくわしくないので何が起こっているのか分かってませんがひとまず。

Comments

[Hayato1201]

ありがとうございます！それが問題でした！

[quickquip]

python setup.py install することで protonets 以下が site-packages にコピーされるだろうと思ったんですが、そうならないんですよね。なぜ?? という感じです。

Answer 2

Python はサブディレクトリにあるモジュールを検索しません。今回のように階層構造になっているモジュールをインポートして使うには、それらをパッケージ化すると便利です。

.
├── main.py
└── protonets
    ├── __init__.py
    ├── engine.py
    └── utils
        ├── __init__.py
        └── data.py

やり方は簡単で、こんな風に各ディレクトリに __init__.py というファイルを作ってみて下さい。これを置くことで Python が各ディレクトリの Python ファイルをパッケージの塊として認識してくれます。

Comments

[Hayato1201]

回答ありがとうございます。
すみません、質問で省略してしまいましたが、こちらのプログラムには元々__init__.pyは入っていました。とすると何が原因でしょうか・・・？

[Hayato1201]

すみません、上記で解決しました！