From dd31a45e8d2143f48c99aa66aad4a8a5b42ff646 Mon Sep 17 00:00:00 2001
From: Akihiro Nitta <nitta@akihironitta.com>
Date: Tue, 8 Dec 2020 06:03:26 +0900
Subject: [PATCH] bugfix: Add missing imports to pl_bolts/__init__.py (#430)

* Add missing imports

* Add missing imports

* Apply isort
---
 pl_bolts/__init__.py | 13 +++++++------
 1 file changed, 7 insertions(+), 6 deletions(-)

diff --git a/pl_bolts/__init__.py b/pl_bolts/__init__.py
index 82a4f14063..40e4c0e6f3 100644
--- a/pl_bolts/__init__.py
+++ b/pl_bolts/__init__.py
@@ -43,15 +43,16 @@
     sys.stdout.write(f'Partial import of `{__name__}` during the build process.\n')  # pragma: no-cover
     # We are not importing the rest of the lightning during the build process, as it may not be compiled yet
 else:
-
-    # from pl_bolts.models.mnist_module import LitMNIST
-    from pl_bolts import callbacks, datamodules, datasets, metrics, models, transforms
+    from pl_bolts import callbacks, datamodules, datasets, losses, metrics, models, optimizers, transforms, utils
 
     __all__ = [
-        # 'LitMNIST',
-        'models',
-        'metrics',
         'callbacks',
         'datamodules',
         'datasets',
+        'losses',
+        'metrics',
+        'models',
+        'optimizers',
+        'transforms',
+        'utils'
     ]