Fix pytorch integration test.

facebookresearch · Jan 25, 2019 · a19dea3 · a19dea3
1 parent eadb3fa
commit a19dea3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/parlai/scripts/build_pytorch_data.py b/parlai/scripts/build_pytorch_data.py
@@ -26,7 +26,7 @@
 
 
 def get_pyt_dict_file(opt):
-    if os.path.exists(opt.get('dict_file', '')):
+    if opt.get('dict_file') and os.path.exists(opt.get('dict_file', '')):
         return opt['dict_file']
     if not opt['pytorch_teacher_task']:
         opt['pytorch_teacher_task'] = opt['task']