Добавить импорт в модуле на лету?

Есть файл `module1.py`.
В нем я сейчас руками пишу:
from root_module import foo

Дальше я динамически импортирую `module1.py`. и вызываю:
module1.foo()

Я хочу сделать так, чтобы в момент динамического импорта в `module1.py` автомтом подсовывалась строка: `from root_module import foo`. Как это сделать?

Сейчас динамический импорт выглядит так:
py_mod = imp.load_source(mod_name, path)
py_mod.foo()