From cf23322c2d13316b3cb6cbe608aab362477391eb Mon Sep 17 00:00:00 2001 From: Gabriel Wen Date: Fri, 14 Feb 2020 15:21:30 -0800 Subject: [PATCH] fix --- xgboost_synthetic/testing/xgboost_test.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/xgboost_synthetic/testing/xgboost_test.py b/xgboost_synthetic/testing/xgboost_test.py index 49602c75c..7becb519a 100644 --- a/xgboost_synthetic/testing/xgboost_test.py +++ b/xgboost_synthetic/testing/xgboost_test.py @@ -99,7 +99,9 @@ def test_xgboost_synthetic(record_xml_attribute, name, namespace, # pylint: disa # Download notebook html to artifacts with open(notebook_artifacts_path, "w") as f: storage_client = storage.Client() - storage_client.download_blob_to_file(output_gcs, f) + bucket = storage_client.get_bucket(nb_bucket) + blob = bucket.get_blob(nb_path) + blob.download_to_file(f) if last_condition.type not in ["Complete"]: logging.error("Job didn't complete successfully")