Save imports

2020-05-19 10:03:35 +02:00
parent 645b7905e8
commit f290d5a8d8
4 changed files with 65 additions and 9 deletions
--- a/visualization/tools.py
+++ b/visualization/tools.py
@ -1,5 +1,10 @@
+try:
+    import matplotlib.pyplot as plt
+except ImportError:  # pragma: no-cover
+    raise ImportError('You want to use `matplotlib` plugins which are not installed yet,'  # pragma: no-cover
+                      ' install it with `pip install matplotlib`.')
+
 from pathlib import Path
-import matplotlib.pyplot as plt


 class Plotter(object):