fix python open issue with non-exist parent dir (#1441)

kubeflow · Jun 5, 2019 · 0ad6c28 · 0ad6c28
1 parent ab97d57
commit 0ad6c28
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/components/kubeflow/katib-launcher/src/launch_study_job.py b/components/kubeflow/katib-launcher/src/launch_study_job.py
@@ -153,6 +153,8 @@ def main(argv=None):
   if wait_response.get("status", {}).get("condition") == "Completed":
     succ = True
     trial = get_best_trial(wait_response["status"]["bestTrialId"])
+    if not os.path.exists(os.path.dirname(args.outputfile)):
+      os.makedirs(os.path.dirname(args.outputfile))
     with open(args.outputfile, 'w') as f:
       ps_dict = {}
       for ps in trial.parameter_set: