fixed y label visibility

sushil79g · Jan 8, 2019 · 5595b63 · 5595b63
1 parent a01050e
commit 5595b63
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/ehtim/image.py b/ehtim/image.py
@@ -2759,8 +2759,9 @@ def display(self, pol=None, cfun='afmhot', interp='gaussian',
         elif label_type=='none':
             plt.axis('off')
             ax = plt.gca()
-            ax.axes.get_xaxis().set_visible(False)
-            ax.axes.get_yaxis().set_visible(False)
+            if axis is None:
+                ax.axes.get_xaxis().set_visible(False)
+                ax.axes.get_yaxis().set_visible(False)
 
         # Show or save to file
         if axis is not None: