Fix the unittests

shunjd · shunjd · commit 7f460e0c75d0 · 2020-01-31T13:08:11.000-08:00
diff --git a/tests/unit/test_sagemaker_steps.py b/tests/unit/test_sagemaker_steps.py
@@ -51,6 +51,10 @@ def pca_estimator():
         mini_batch_size=200
     )
 
+    pca.sagemaker_session = MagicMock()
+    pca.sagemaker_session.boto_region_name = 'us-east-1'
+    pca.sagemaker_session._default_bucket = 'sagemaker'
+
     return pca
 
 @pytest.fixture

Original file line number	Diff line number	Diff line change
`@@ -51,6 +51,10 @@ def pca_estimator():`
`51`	`51`	`mini_batch_size=200`
`52`	`52`	`)`
`53`	`53`
	`54`	`+ pca.sagemaker_session = MagicMock()`
	`55`	`+ pca.sagemaker_session.boto_region_name = 'us-east-1'`
	`56`	`+ pca.sagemaker_session._default_bucket = 'sagemaker'`
	`57`	`+`
`54`	`58`	`return pca`
`55`	`59`
`56`	`60`	`@pytest.fixture`