fix: matplotlib import to inside function

Author: satra

nipype/interfaces/nipy/model.py

@@ -4,11 +4,6 @@
 import nibabel as nb
 import numpy as np
 
-pylab_available = True
-try:
-    import pylab
-except:
-    pylab_available = False
 
 from ...utils.misc import package_check
 
@@ -140,13 +135,11 @@ def _run_interface(self, runtime):
                 design_matrix[:,i] = (design_matrix[:,i]-design_matrix[:,i].mean())/design_matrix[:,i].std()
 
         if self.inputs.plot_design_matrix:
-            if pylab_available:
-                pylab.pcolor(design_matrix)
-                pylab.savefig("design_matrix.pdf")
-                pylab.close()
-                pylab.clf()
-            else:
-                Exception('Pylab not available for saving design matrix image')
+            import pylab
+            pylab.pcolor(design_matrix)
+            pylab.savefig("design_matrix.pdf")
+            pylab.close()
+            pylab.clf()
 
         glm = GLM.glm()
         glm.fit(timeseries.T, design_matrix, method=self.inputs.method, model=self.inputs.model)