mkdirs

kubeflow · Feb 14, 2020 · 480fa8a · 480fa8a
1 parent dcfb493
commit 480fa8a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/xgboost_synthetic/testing/xgboost_test.py b/xgboost_synthetic/testing/xgboost_test.py
@@ -95,6 +95,7 @@ def test_xgboost_synthetic(record_xml_attribute, name, namespace, # pylint: disa
   with open(os.path.join(notebook_artifacts_dir, "notebook.html"), "w") as f:
   notebook_artifacts_path = os.path.join(notebook_artifacts_dir, "notebook.html")
   logging.info("Writing notebook artifact to: %s", notebook_artifacts_path)
+  os.makedirs(notebook_artifacts_dir, exist_ok=True)
   # Download notebook html to artifacts
   with open(notebook_artifacts_path, "w") as f:
     storage_client = storage.Client()