Как заставить IntelliJ IDEA распознавать каталог Python в мультипроекте Java Gradle?

Question

У меня есть мультипроект Java, который также содержит некоторые python сценарии сборки:

my_project/
├── build.gradle.kts
├── settings.gradle.kts
├── module_a/
│   ├── build.gradle.kts
│   └── src/main/java/
├── module_b/
│   ├── build.gradle.kts
│   └── src/main/java/
└── buildscript/
    ├── build.py
    ├── common_func.py
    └── tests/
        ├── test_build.py
        └── test_common_func.py

IntelliJ распознает каталог buildscript как Python, если я вручную отмечу его как каталог источника. Затем я получаю завершение кода и могу перемещаться по коду. Я также могу нормально запускать тесты.

Но всякий раз, когда я заново импортирую проект Gradle, IDEA забывает о каталоге buildscript, и мне приходится вручную отмечать его как исходный. Что я могу сделать, чтобы это исправить? Как я полагаю, есть некоторые настройки Gradle?

Дополнительная информация о тестах: у меня есть sys.path.append(os.path.dirname(__file__) + '/..') вверху тестовых файлов, и у меня есть задача Exe c для них:

tasks.register<Exec>("testBuildScripts") {
    workingDir(rootDir)
    commandLine("python3", "-m", "unittest", "discover", "buildscript/tests/")
}