Set matplotlib backend from config in pyscript

Author: mwaskom

nipype/pipeline/plugins/base.py

@@ -91,13 +91,6 @@ def report_nodes_not_run(notrun):
  'Check log for details'))
 
 
-def write_matplotlibrc(batch_dir):
- """Set the matplotlib backend in a local rc file."""
- backend = nipype.config.get("execution", "matplotlib_backend")
- with open(os.path.join(batch_dir, "matplotlibrc"), "w") as fid:
- fid.write("backend : %s" % backend)
-
-
 def create_pyscript(node, updatehash=False, store_exception=True):
  # pickle node
  timestamp = strftime('%Y%m%d_%H%M%S')
@@ -111,12 +104,14 @@ def create_pyscript(node, updatehash=False, store_exception=True):
  batch_dir = os.path.join(node.base_dir, 'batch')
  if not os.path.exists(batch_dir):
  os.makedirs(batch_dir)
- write_matplotlibrc(batch_dir)
  pkl_file = os.path.join(batch_dir, 'node_%s.pklz' % suffix)
  savepkl(pkl_file, dict(node=node, updatehash=updatehash))
+ mpl_backend = nipype.config.get("execution", "matplotlib_backend")
  # create python script to load and trap exception
  cmdstr = """import os
 import sys
+import matplotlib
+matplotlib.use('%s')
 from nipype import config, logging
 from nipype.utils.filemanip import loadpkl, savepkl
 from socket import gethostname
@@ -162,7 +157,7 @@ def create_pyscript(node, updatehash=False, store_exception=True):
  report_crash(info['node'], traceback, gethostname())
  raise Exception(e)
 """
- cmdstr = cmdstr % (pkl_file, batch_dir, node.config, suffix)
+ cmdstr = cmdstr % (mpl_backend, pkl_file, batch_dir, node.config, suffix)
  pyscript = os.path.join(batch_dir, 'pyscript_%s.py' % suffix)
  fp = open(pyscript, 'wt')
  fp.writelines(cmdstr)