|
import importlib |
|
|
|
__modules__ = {} |
|
|
|
|
|
def register(name): |
|
def decorator(cls): |
|
if name in __modules__: |
|
raise ValueError( |
|
f"Module {name} already exists! Names of extensions conflict!" |
|
) |
|
else: |
|
__modules__[name] = cls |
|
return cls |
|
|
|
return decorator |
|
|
|
|
|
def find(name): |
|
if name in __modules__: |
|
return __modules__[name] |
|
else: |
|
try: |
|
module_string = ".".join(name.split(".")[:-1]) |
|
cls_name = name.split(".")[-1] |
|
module = importlib.import_module(module_string, package=None) |
|
return getattr(module, cls_name) |
|
except Exception as e: |
|
raise ValueError(f"Module {name} not found!") |
|
|
|
|
|
|
|
import logging |
|
|
|
logger = logging.getLogger("pytorch_lightning") |
|
|
|
from pytorch_lightning.utilities.rank_zero import ( |
|
rank_zero_debug, |
|
rank_zero_info, |
|
rank_zero_only, |
|
) |
|
|
|
debug = rank_zero_debug |
|
info = rank_zero_info |
|
|
|
|
|
@rank_zero_only |
|
def warn(*args, **kwargs): |
|
logger.warn(*args, **kwargs) |
|
|
|
|
|
from . import data, models, systems |
|
|
