add metrics to pl module (#2506)

Lightning-AI · Jul 5, 2020 · b6507da · b6507da
1 parent 9924c76
commit b6507da
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pytorch_lightning/__init__.py b/pytorch_lightning/__init__.py
@@ -54,13 +54,15 @@
     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',
         'LightningModule',
         'Callback',
         'data_loader',
         'seed_everything',
+        'metrics'
     ]
 
     # necessary for regular bolts imports. Skip exception since bolts is not always installed