Skip to content

Commit 8c379c1

Browse files
committed
chore: assume kms support if 1 instance in group supports kms
1 parent d42cad6 commit 8c379c1

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/sagemaker/estimator.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -618,7 +618,7 @@ def __init__(
618618
instance_type_for_volume_kms and instance_supports_kms(instance_type_for_volume_kms)
619619
)
620620
or instance_groups is not None
621-
and all(
621+
and any(
622622
[
623623
instance_supports_kms(instance_group.instance_type)
624624
for instance_group in instance_groups

src/sagemaker/session.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -3862,7 +3862,7 @@ def create_endpoint_config_from_existing(
38623862
if new_kms_key is not None or existing_endpoint_config_desc.get("KmsKeyId") is not None:
38633863
request["KmsKeyId"] = new_kms_key or existing_endpoint_config_desc.get("KmsKeyId")
38643864

3865-
supports_kms = all(
3865+
supports_kms = any(
38663866
[
38673867
instance_supports_kms(production_variant["InstanceType"])
38683868
for production_variant in production_variants
@@ -4492,7 +4492,7 @@ def endpoint_from_production_variants(
44924492
str: The name of the created ``Endpoint``.
44934493
"""
44944494

4495-
supports_kms = all(
4495+
supports_kms = any(
44964496
[
44974497
instance_supports_kms(production_variant["InstanceType"])
44984498
for production_variant in production_variants

0 commit comments

Comments
 (0)