bugfix: Add missing imports to pl_bolts/__init__.py (Lightning-Univer…

…se#430)

* Add missing imports

* Add missing imports

* Apply isort

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'
     ]