Merge pull request #39 from njsmith/fix-save-option

Fix --save
matplotlib · Jun 26, 2018 · 634fffa · 634fffa
2 parents 17feb91 + 5bd78e7
commit 634fffa
Showing 1 changed file with 6 additions and 5 deletions.
diff --git a/viscm/gui.py b/viscm/gui.py
@@ -1017,8 +1017,8 @@ def main(argv):
         v = viscm(cm.cmap, name=cm.name, figure=fig, uniform_space=cm.uniform_space)
         mainwindow = ViewerWindow(figureCanvas, v, cm.name)
         if args.save is not None:
-            v.fig.set_size_inches(20, 12)
-            v.fig.savefig(args.save)
+            v.figure.set_size_inches(20, 12)
+            v.figure.savefig(args.save)
     elif args.action == "edit":
         if not cm.can_edit:
             sys.exit("Sorry, I don't know how to edit the specified colormap")
@@ -1305,9 +1305,10 @@ def closeEvent(self, ce):
         self.fileQuit()
 
     def save(self):
-        fileName = QtWidgets.QFileDialog.getSaveFileName(caption="Save file",
-                                                     directory=self.viscm_editor.name + ".jscm",
-                                                     filter="JSCM Files (*.jscm)")
+        fileName = QtWidgets.QFileDialog.getSaveFileName(
+            caption="Save file",
+            directory=self.viscm_editor.name + ".jscm",
+            filter="JSCM Files (*.jscm)")
         self.viscm_editor.save_colormap(fileName)
 
     def loadviewer(self):