Skip to content

Commit 29c9728

Browse files
balajisankar15Balaji Sankar
authored andcommitted
fix: Make 'role' as Optional for ModelQualityMonitor and DefaultModelMonitor, and fixed PROCESSING_CONFIG_PATH (aws#849)
Co-authored-by: Balaji Sankar <[email protected]>
1 parent 669e5a6 commit 29c9728

File tree

3 files changed

+4
-6
lines changed

3 files changed

+4
-6
lines changed

src/sagemaker/model_monitor/clarify_model_monitoring.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -40,7 +40,7 @@ class ClarifyModelMonitor(mm.ModelMonitor):
4040

4141
def __init__(
4242
self,
43-
role,
43+
role=None,
4444
instance_count=1,
4545
instance_type="ml.m5.xlarge",
4646
volume_size_in_gb=30,

src/sagemaker/model_monitor/model_monitoring.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1565,7 +1565,7 @@ class DefaultModelMonitor(ModelMonitor):
15651565

15661566
def __init__(
15671567
self,
1568-
role,
1568+
role=None,
15691569
instance_count=1,
15701570
instance_type="ml.m5.xlarge",
15711571
volume_size_in_gb=30,
@@ -2632,7 +2632,7 @@ class ModelQualityMonitor(ModelMonitor):
26322632

26332633
def __init__(
26342634
self,
2635-
role,
2635+
role=None,
26362636
instance_count=1,
26372637
instance_type="ml.m5.xlarge",
26382638
volume_size_in_gb=30,

src/sagemaker/session.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -242,9 +242,7 @@ def _simple_path(*args: str):
242242
PROCESSING_JOB_SECURITY_GROUP_IDS_PATH = _simple_path(
243243
PROCESSING_JOB_VPC_CONFIG_PATH, SECURITY_GROUP_IDS
244244
)
245-
PROCESSING_OUTPUT_CONFIG_PATH = _simple_path(
246-
SAGEMAKER, PROCESSING_JOB, PROCESSING_OUTPUT_CONFIG, KMS_KEY_ID
247-
)
245+
PROCESSING_OUTPUT_CONFIG_PATH = _simple_path(SAGEMAKER, PROCESSING_JOB, PROCESSING_OUTPUT_CONFIG)
248246
PROCESSING_JOB_KMS_KEY_ID_PATH = _simple_path(
249247
SAGEMAKER, PROCESSING_JOB, PROCESSING_OUTPUT_CONFIG, KMS_KEY_ID
250248
)

0 commit comments

Comments
 (0)