When Installed from source, pylance has difficulties to resolve the imports

[sanggusti]

Bug description

Expected:
When installed via pip install -e ".[all]", the project could be imported and the interface on any IDE would be seamless,

What happened:
pylance in vscode couldn't resolve the imports. My workaround need to define extra paths to litgpt folder. Seems like the bug on setup file.

Thoughts:
Lightning projects usually have file structure to be root_project/src/library_name, and litgpt has litgpt/litgpt file structure. pyproject.toml file in litgpt is also following the structure in other lightning project. I think this structures need to be standardized or some adjustments in pyproject.toml need to be done.

Reproduced in studio

No response

What operating system are you using?

Linux

LitGPT Version

litgpt-0.5.12

[sanggusti]

How about to include setup.py files in the project root so the installation via source could be more configurable?

[bhimrazy]

Hi @sanggusti,

It looks like the all extra doesn’t exist in this project, so the installation may not have completed as expected.

Could you try installing with the explicit extras instead?

pip install -e ".[extra,compiler,test]"

Or, if you’re using uv:

uv sync --all-extras

Please let us know if this resolves the issue, or if Pylance still has trouble resolving the imports after a clean install.