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 1919from  mock  import  patch , Mock , ANY 
2020
2121from  sagemaker .config  import  load_sagemaker_config 
22+ from  sagemaker .session_settings  import  SessionSettings 
2223from  tests .unit  import  DATA_DIR 
2324from  sagemaker .remote_function .job  import  (
2425    _JobSettings ,
@@ -98,7 +99,7 @@ def mock_session():
9899    session  =  Mock ()
99100    session .sagemaker_client .create_training_job .return_value  =  {"TrainingJobArn" : TRAINING_JOB_ARN }
100101    session .sagemaker_client .describe_training_job .return_value  =  COMPLETED_TRAINING_JOB 
101- 
102+      session . settings   =   SessionSettings () 
102103    session .default_bucket .return_value  =  BUCKET 
103104    session .expand_role .return_value  =  ROLE_ARN 
104105    session .boto_region_name  =  TEST_REGION 
Original file line number Diff line number Diff line change 2020from  mock  import  Mock , patch 
2121
2222from  sagemaker  import  s3 
23+ from  sagemaker .session_settings  import  SessionSettings 
2324from  sagemaker .workflow .condition_step  import  ConditionStep 
2425from  sagemaker .workflow .conditions  import  ConditionEquals 
2526from  sagemaker .workflow .execution_variables  import  ExecutionVariables 
@@ -79,6 +80,7 @@ def test_pipeline_create_and_update_with_config_injection(sagemaker_session_mock
7980    sagemaker_session_mock .sagemaker_client .describe_pipeline .return_value  =  {
8081        "PipelineArn" : "pipeline-arn" 
8182    }
83+     sagemaker_session_mock .settings  =  SessionSettings ()
8284    pipeline  =  Pipeline (
8385        name = "MyPipeline" ,
8486        parameters = [],
    
 
   
 
     
   
   
          
     
  
    
     
 
    
      
     
 
     
    You can’t perform that action at this time.
  
 
    
  
     
    
      
        
     
 
       
      
     
   
 
    
    
  
 
  
 
     
    
0 commit comments