Как импортировать пользовательские файлы в ту же папку, что и main.py ⇐ Python

[Anonymous]

Предположим, у меня есть:

Код: Выделить всё

Project_root (folder)
source (folder)
main.py
functions(folder)
test.py
__init__.py (empty)

Вот код в main.py:

Код: Выделить всё

import os
import sys
from source.functions import test
test.hello()

Код в test.py:

Код: Выделить всё

def hello():
print("Hello World!")

Я запускаю это на коде vs в виртуальной среде anaconda. Я использую плагин tensorflow-directml.
Я получаю следующую ошибку:

Код: Выделить всё

line 4, in 
from source.functions import test
ModuleNotFoundError: No module named 'source'

Я пытался импортировать оба файла Python в одной папке, и это сработало. Мне также удалось, когда main.py находился непосредственно в корневой папке, а не в папке, которая находилась ПОД корневой папкой. В этом случае файл main.py находится в корневой папке и происходит сбой.

Подробнее здесь: https://stackoverflow.com/questions/786 ... as-main-py