Skip to content

Commit 2f96df3

Browse files
authored
Merge branch 'master' into updated_js_mb_compression_logic
2 parents f7ecefb + 889d7b7 commit 2f96df3

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

41 files changed

+1736
-864
lines changed

src/sagemaker/estimator.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -578,6 +578,17 @@ def __init__(
578578
self.dependencies = dependencies or []
579579
self.uploaded_code: Optional[UploadedCode] = None
580580

581+
# Check that the user properly sets both subnet and secutiry_groupe_ids
582+
if (
583+
subnets is not None
584+
and security_group_ids is None
585+
or security_group_ids is not None
586+
and subnets is None
587+
):
588+
raise RuntimeError(
589+
"When setting up custom VPC, both subnets and security_group_ids must be set"
590+
)
591+
581592
if self.instance_type in ("local", "local_gpu"):
582593
if self.instance_type == "local_gpu" and self.instance_count > 1:
583594
raise RuntimeError("Distributed Training in Local GPU is not supported")

0 commit comments

Comments
 (0)