fixed smallcorgi#69

wadefelix · Dec 16, 2016 · 80a0080 · 80a0080
1 parent 284ecbb
commit 80a0080
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tools/gapttest.py b/tools/gapttest.py
@@ -246,10 +246,10 @@ def makeEntry(self, name, default='', cmd=None):
 
     def askopenfilename_model(self):
         filename = tkFileDialog.askopenfilename(initialdir = self.initdir, filetypes = [('tensorflow checkpoint file', '.ckpt'), ('tensorflow checkpoint meta files', '.ckpt.meta'), ('all files', '*')])
-        if filename.endswith('.ckpt.meta'):
+        if filename and filename.endswith('.ckpt.meta'):
             filename = filename[:-5]
-        self.textvariables['model'].set(filename)
-        self.initdir = os.path.dirname(filename)
+            self.textvariables['model'].set(filename)
+            self.initdir = os.path.dirname(filename)
     def close(self):
         self.quit()
     def resize(self, event):