Add missing optional packages to requirements/*.txt (Lightning-Univ…

…erse#450)

* Import matplotlib at the top

* Add missing optional packages

* Update wandb

* Add mypy to requirements

pl_bolts/callbacks/vision/confused_logit.py

@@ -1,7 +1,17 @@
+import importlib
+
 import torch
 from pytorch_lightning import Callback
 from torch import nn
 
+from pl_bolts.utils.warnings import warn_missing_pkg
+
+_MATPLOTLIB_AVAILABLE = importlib.util.find_spec("matplotlib") is not None
+if _MATPLOTLIB_AVAILABLE:
+    from matplotlib import pyplot as plt
+else:
+    warn_missing_pkg("matplotlib")  # pragma: no-cover
+
 
 class ConfusedLogitCallback(Callback):  # pragma: no-cover
     """
@@ -93,7 +103,10 @@ def training_step(...):
                 pl_module.train()
 
     def _plot(self, confusing_x, confusing_y, trainer, model, mask_idxs):
-        from matplotlib import pyplot as plt
+        if not _MATPLOTLIB_AVAILABLE:
+            raise ModuleNotFoundError(  # pragma: no-cover
+                'You want to use `matplotlib` which is not installed yet, install it with `pip install matplotlib`.'
+            )
 
         confusing_x = confusing_x[:self.top_k]
         confusing_y = confusing_y[:self.top_k]

requirements/loggers.txt

@@ -1,2 +1,5 @@
 # test_tube>=0.7.5
-# trains>=0.14.1
+# trains>=0.14.1
+matplotlib
+wandb
+scipy

requirements/test.txt

@@ -9,5 +9,6 @@ check-manifest
 twine==1.13.0
 isort>=5.6.4
 pre-commit>=1.0
+mypy
 
 atari-py==0.2.6  # needed for RL