lightning/pytorch_lightning/__init__.py

"""Root package info."""

import logging
import os

from pytorch_lightning.__about__ import *  # noqa: F401 F403

_root_logger = logging.getLogger()
_logger = logging.getLogger(__name__)
_logger.setLevel(logging.INFO)

# if root logger has handlers, propagate messages up and let root logger process them
if not _root_logger.hasHandlers():
    _logger.addHandler(logging.StreamHandler())
    _logger.propagate = False

_PACKAGE_ROOT = os.path.dirname(__file__)
_PROJECT_ROOT = os.path.dirname(_PACKAGE_ROOT)

from pytorch_lightning import metrics  # noqa: E402
from pytorch_lightning.callbacks import Callback  # noqa: E402
from pytorch_lightning.core import LightningDataModule, LightningModule  # noqa: E402
from pytorch_lightning.trainer import Trainer  # noqa: E402
from pytorch_lightning.utilities.seed import seed_everything  # noqa: E402

__all__ = [
    'Trainer',
    'LightningDataModule',
    'LightningModule',
    'Callback',
    'seed_everything',
    'metrics',
]

# for compatibility with namespace packages
__import__('pkg_resources').declare_namespace(__name__)