fix for save arg parsing

59dc8dff · Jameson Rollins · a101f0c4 · 59dc8dff
Commit 59dc8dff authored Jan 05, 2021 by Jameson Rollins
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

gwinc/__main__.py gwinc/__main__.py +2 -2

No files found.
--- a/gwinc/__main__.py
+++ b/gwinc/__main__.py
@@ -181,11 +181,11 @@ def main():
                print(fmt.format(k, v, ov))
        return

-    out_files = set(args.save)
    out_data_files = set()
    out_plot_files = set()
-    if out_files:
+    if args.save:
        args.plot = False
+        out_files = set(args.save)
        for path in out_files:
            if os.path.splitext(path)[1] in io.DATA_SAVE_FORMATS:
                out_data_files.add(path)