Skip to content

Commit 55e93d9

Browse files
committed
fix: failing unit tests
1 parent fc4096b commit 55e93d9

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

tests/unit/sagemaker/remote_function/test_job.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
from mock import patch, Mock, ANY
2020

2121
from sagemaker.config import load_sagemaker_config
22+
from sagemaker.session_settings import SessionSettings
2223
from tests.unit import DATA_DIR
2324
from 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

tests/unit/sagemaker/workflow/test_pipeline.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
from mock import Mock, patch
2121

2222
from sagemaker import s3
23+
from sagemaker.session_settings import SessionSettings
2324
from sagemaker.workflow.condition_step import ConditionStep
2425
from sagemaker.workflow.conditions import ConditionEquals
2526
from 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=[],

0 commit comments

Comments
 (0)