diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py index 9a6469fc72..a03f2b9e80 100644 --- a/pytorch_lightning/__init__.py +++ b/pytorch_lightning/__init__.py @@ -54,6 +54,7 @@ else: from pytorch_lightning.callbacks import Callback from pytorch_lightning.trainer import Trainer from pytorch_lightning.utilities.seed import seed_everything + from pytorch_lightning import metrics __all__ = [ 'Trainer', @@ -61,6 +62,7 @@ else: 'Callback', 'data_loader', 'seed_everything', + 'metrics' ] # necessary for regular bolts imports. Skip exception since bolts is not always installed