RF: Make matplotlib an optional import.

dipy · Aug 4, 2012 · d0ed95e · d0ed95e
1 parent 898a401
commit d0ed95e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/dipy/viz/projections.py b/dipy/viz/projections.py
@@ -8,9 +8,14 @@
 import numpy as np
 import scipy.interpolate as interp
 
-import matplotlib
-import matplotlib.pyplot as plt
-import matplotlib.tri as tri
+try:
+    import matplotlib
+    import matplotlib.pyplot as plt
+    import matplotlib.tri as tri
+except ImportError:
+    e_s = "You do not have Matplotlib installed. Some visualization functions"
+    e_s += " might not work for you."
+    print(e_s)
 
 import dipy.core.geometry as geo