"""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.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"] # for compatibility with namespace packages __import__("pkg_resources").declare_namespace(__name__)