fix: use default bucket for checkpoint_s3_uri

Chuyang Deng · Chuyang Deng · commit 36c9ac667fe1 · 2019-10-11T15:35:59.000-07:00
diff --git a/tests/integ/test_tf_script_mode.py b/tests/integ/test_tf_script_mode.py
@@ -71,7 +71,7 @@ def test_mnist(sagemaker_session, instance_type):
     reason="checkpoint s3 bucket is in us-east-1, ListObjectsV2 will fail in other regions",
 )
 def test_checkpoint_config(sagemaker_session, instance_type):
-    checkpoint_s3_uri = "s3://142577830533-us-east-1-sagemaker-checkpoint"
+    checkpoint_s3_uri = "s3://{}".format(sagemaker_session.default_bucket())
     checkpoint_local_path = "/test/checkpoint/path"
     estimator = TensorFlow(
         entry_point=SCRIPT,

Original file line number	Diff line number	Diff line change
`@@ -71,7 +71,7 @@ def test_mnist(sagemaker_session, instance_type):`
`71`	`71`	`reason="checkpoint s3 bucket is in us-east-1, ListObjectsV2 will fail in other regions",`
`72`	`72`	`)`
`73`	`73`	`def test_checkpoint_config(sagemaker_session, instance_type):`
`74`		`- checkpoint_s3_uri = "s3://142577830533-us-east-1-sagemaker-checkpoint"`
	`74`	`+ checkpoint_s3_uri = "s3://{}".format(sagemaker_session.default_bucket())`
`75`	`75`	`checkpoint_local_path = "/test/checkpoint/path"`
`76`	`76`	`estimator = TensorFlow(`
`77`	`77`	`entry_point=SCRIPT,`