File tree Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Expand file tree Collapse file tree 2 files changed +4
-1
lines changed Original file line number Diff line number Diff line change 19
19
from mock import patch , Mock , ANY
20
20
21
21
from sagemaker .config import load_sagemaker_config
22
+ from sagemaker .session_settings import SessionSettings
22
23
from tests .unit import DATA_DIR
23
24
from sagemaker .remote_function .job import (
24
25
_JobSettings ,
@@ -98,7 +99,7 @@ def mock_session():
98
99
session = Mock ()
99
100
session .sagemaker_client .create_training_job .return_value = {"TrainingJobArn" : TRAINING_JOB_ARN }
100
101
session .sagemaker_client .describe_training_job .return_value = COMPLETED_TRAINING_JOB
101
-
102
+ session . settings = SessionSettings ()
102
103
session .default_bucket .return_value = BUCKET
103
104
session .expand_role .return_value = ROLE_ARN
104
105
session .boto_region_name = TEST_REGION
Original file line number Diff line number Diff line change 20
20
from mock import Mock , patch
21
21
22
22
from sagemaker import s3
23
+ from sagemaker .session_settings import SessionSettings
23
24
from sagemaker .workflow .condition_step import ConditionStep
24
25
from sagemaker .workflow .conditions import ConditionEquals
25
26
from sagemaker .workflow .execution_variables import ExecutionVariables
@@ -79,6 +80,7 @@ def test_pipeline_create_and_update_with_config_injection(sagemaker_session_mock
79
80
sagemaker_session_mock .sagemaker_client .describe_pipeline .return_value = {
80
81
"PipelineArn" : "pipeline-arn"
81
82
}
83
+ sagemaker_session_mock .settings = SessionSettings ()
82
84
pipeline = Pipeline (
83
85
name = "MyPipeline" ,
84
86
parameters = [],
You can’t perform that action at this time.
0 commit comments