あるはずのディレクトリ なのにModuleNotFoundErrorが出る

Question

sample_policy.pyというファイルを走らせたいのですが、importの部分でModuleNotFoundErrorが出ます。worldmodels2というディレクトリ からファイルを呼んでいて、worldmodelsというディレクトリ はすぐ下にあるのですが、何が間違っているのでしょうか？

エラー

(venv) home:~/worldmodels2$ python3 worldmodels/data/sample_policy.py 
Traceback (most recent call last):
  File "worldmodels/data/sample_policy.py", line 10, in <module>
    from worldmodels.control.train_controller import episode
ModuleNotFoundError: No module named 'worldmodels'

ディレクトリの構成

worldmodels2
└── worldmodels
    ├── __init__.py
    ├── control
    │   ├── controller.py
    │   └── train_controller.py
    ├── data
    │   ├── __init__.py
    │   ├── car_racing.py
    │   ├── sample_latent_stats.py
    │   ├── sample_policy.py
    │   └── tf_records.py
    ├── memory
    │   ├── __init__.py
    │   ├── memory.py
    │   └── train_memory.py
    ├── params.py
    ├── tests
    │   ├── test_gif_creation.py
    │   ├── test_starmap.py
    │   └── test_tf_records.py
    ├── utils.py
    └── vision
        ├── __init__.py
        ├── images.py
        ├── train_vae.py
        └── vae.py
```

Answer 1

ソースがないので 推測ですが

ホームディレクトリ配下にあるようで、ホームディレクトリから起動するとすれば
メインの最初あたりに

import os
import sys; sys.path = [os.getcwd()] + sys.path

をいれれば、パスが通るようになるのでは？

Answer 2

そうやって実行すると~/worldmodels2/worldmodels/data/がモジュール探索の基準ディレクトリになってしまいます。

実行するpyファイルはトップ（ここでいう~/worldmodels2/）に置くように徹底するか、もしくは、

cd `~/worldmodels2/
python3 -m worldmodels.data.sample_policy

と実行するかします。
https://docs.python.org/ja/3/using/cmdline.html#cmdoption-m