# HG changeset patch
# User Nicolas Saunier <nicolas.saunier@polymtl.ca>
# Date 1469051560 14400
# Node ID 36c5bee9a88777e4110439c61c3d32913bf0f974
# Parent  37d14818db8999964c1ee78e595d46a56e6ea53c
bug correction

diff -r 37d14818db89 -r 36c5bee9a887 python/utils.py
--- a/python/utils.py	Fri Jul 15 16:31:37 2016 -0400
+++ b/python/utils.py	Wed Jul 20 17:52:40 2016 -0400
@@ -350,7 +350,10 @@
         fig.add_subplot(1,1,1)
         #plt.imshow(np.fabs(c), interpolation='none')
         plt.imshow(c, vmin=-1., vmax = 1., interpolation='none', cmap = 'RdYlBu_r') # coolwarm
-        colnames = [displayNames.get(s.strip(), s.strip()) for s in columns]
+        if displayNames is not None:
+            colnames = [displayNames.get(s.strip(), s.strip()) for s in columns]
+        else:
+            colnames = columns
         #correlation.plot_corr(c, xnames = colnames, normcolor=True, title = filename)
         plt.xticks(range(len(colnames)), colnames, rotation=90)
         plt.yticks(range(len(colnames)), colnames)