pytorch-lightning

1
"""Root package info."""
2

3
import logging
4
import os
5

6
from lightning_utilities.core.imports import module_available, package_available
7

8
_root_logger = logging.getLogger()
9
_logger = logging.getLogger(__name__)
10
_logger.setLevel(logging.INFO)
11

12
_console = logging.StreamHandler()
13
_console.setLevel(logging.INFO)
14

15
formatter = logging.Formatter("%(levelname)s: %(message)s")
16
_console.setFormatter(formatter)
17

18
# if root logger has handlers, propagate messages up and let root logger process them,
19
# otherwise use our own handler
20
if not _root_logger.hasHandlers():
21
    _logger.addHandler(_console)
22
    _logger.propagate = False
23

24

25
if os.path.isfile(os.path.join(os.path.dirname(__file__), "__about__.py")):
26
    from lightning.app.__about__ import *  # noqa: F403
27
if "__version__" not in locals():
28
    if os.path.isfile(os.path.join(os.path.dirname(__file__), "__version__.py")):
29
        from lightning.app.__version__ import version as __version__
30
    elif package_available("lightning"):
31
        from lightning import __version__  # noqa: F401
32

33
from lightning.app.core.app import LightningApp  # noqa: E402
34
from lightning.app.core.flow import LightningFlow  # noqa: E402
35
from lightning.app.core.work import LightningWork  # noqa: E402
36
from lightning.app.plugin.plugin import LightningPlugin  # noqa: E402
37
from lightning.app.utilities.packaging.build_config import BuildConfig  # noqa: E402
38
from lightning.app.utilities.packaging.cloud_compute import CloudCompute  # noqa: E402
39

40
if module_available("lightning.app.components.demo"):
41
    from lightning.app.components import demo  # noqa: F401
42

43
__package_name__ = "lightning.app".split(".")[0]
44

45
_PACKAGE_ROOT = os.path.dirname(__file__)
46
_PROJECT_ROOT = os.path.dirname(os.path.dirname(_PACKAGE_ROOT))
47
if __package_name__ == "lightning":
48
    _PACKAGE_ROOT = os.path.dirname(_PACKAGE_ROOT)
49
    _PROJECT_ROOT = os.path.dirname(_PROJECT_ROOT)
50

51
__all__ = ["LightningApp", "LightningFlow", "LightningWork", "LightningPlugin", "BuildConfig", "CloudCompute"]
52