From dcc12c31b79ac9a17738d75adba65556624c412e Mon Sep 17 00:00:00 2001
From: Vee9ahd1 <Vee9ahd1@arson.club>
Date: Mon, 20 May 2019 23:24:07 -0400
Subject: prefix bug fix

---
 gantools/cli.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'gantools')

diff --git a/gantools/cli.py b/gantools/cli.py
index d15e01d..73213ed 100644
--- a/gantools/cli.py
+++ b/gantools/cli.py
@@ -38,7 +38,8 @@ def main():
     ims = gan.sample(z_seq, label_seq, truncation_seq, args.nbatch)
 
     # save images to file
-    print('Saving image files: '+str(args.pathprefix))
-    image_utils.save_images(ims, args.pathprefix)
+    pathprefix = '' if args.pathprefix == None else str(args.pathprefix)
+    print('Saving image files: '+pathprefix)
+    image_utils.save_images(ims, pathprefix)
 
     print('Done.')
-- 
cgit v1.2.1