diff --git a/.github/workflows/ci-tests-store.yml b/.github/workflows/ci-tests-store.yml index ba05dfb9d6d73..52949009d70df 100644 --- a/.github/workflows/ci-tests-store.yml +++ b/.github/workflows/ci-tests-store.yml @@ -10,6 +10,11 @@ on: paths: - ".actions/*" - "requirements/store/**" + - "src/lightning/__about__.py" + - "src/lightning/__init__.py" + - "src/lightning/__main__.py" + - "src/lightning/__setup__.py" + - "src/lightning/__version__.py" - "src/lightning/store/**" - "tests/tests_store/**" - "pyproject.toml" # includes pytest config diff --git a/src/lightning/__init__.py b/src/lightning/__init__.py index 385af0de771e7..b506f72ed1d87 100644 --- a/src/lightning/__init__.py +++ b/src/lightning/__init__.py @@ -1,6 +1,5 @@ """Root package info.""" import logging -import os # explicitly don't set root logger's propagation and leave this to subpackages to manage _logger = logging.getLogger(__name__) @@ -29,9 +28,7 @@ from lightning.pytorch.trainer import Trainer # noqa: E402 import lightning.app # isort: skip # noqa: E402 -import lightning.store # isort: skip # noqa: E402 - -lightning.app._PROJECT_ROOT = os.path.dirname(lightning.app._PROJECT_ROOT) +import lightning.store # isort: skip # noqa: E402, F401 __all__ = [ diff --git a/src/lightning/app/__init__.py b/src/lightning/app/__init__.py index 13983ef64e101..6e4635c83c48b 100644 --- a/src/lightning/app/__init__.py +++ b/src/lightning/app/__init__.py @@ -45,5 +45,8 @@ _PACKAGE_ROOT = os.path.dirname(__file__) _PROJECT_ROOT = os.path.dirname(os.path.dirname(_PACKAGE_ROOT)) +if __package_name__ == "lightning": + _PACKAGE_ROOT = os.path.dirname(_PACKAGE_ROOT) + _PROJECT_ROOT = os.path.dirname(_PROJECT_ROOT) __all__ = ["LightningApp", "LightningFlow", "LightningWork", "LightningPlugin", "BuildConfig", "CloudCompute"]