Skip to content

Commit aa99b5b

Browse files
viclzhusagemaker-bot
authored andcommitted
deprecation: remove explicit partitions key requirement on pysdk side. (aws#4057)
1 parent 40ec1c8 commit aa99b5b

File tree

2 files changed

+0
-9
lines changed

2 files changed

+0
-9
lines changed

src/sagemaker/fw_utils.py

-3
Original file line numberDiff line numberDiff line change
@@ -327,9 +327,6 @@ def validate_mp_config(config):
327327
ValueError: If any of the keys have incorrect values.
328328
"""
329329

330-
if "partitions" not in config:
331-
raise ValueError("'partitions' is a required parameter.")
332-
333330
def validate_positive(key):
334331
try:
335332
if not isinstance(config[key], int) or config[key] < 1:

tests/unit/test_fw_utils.py

-6
Original file line numberDiff line numberDiff line change
@@ -135,12 +135,6 @@ def test_tar_and_upload_dir_s3_without_kms_with_overridden_settings(utils, sagem
135135
obj.upload_file.assert_called_with(utils.create_tar_file(), ExtraArgs=None)
136136

137137

138-
def test_mp_config_partition_exists():
139-
mp_parameters = {}
140-
with pytest.raises(ValueError):
141-
fw_utils.validate_mp_config(mp_parameters)
142-
143-
144138
@pytest.mark.parametrize(
145139
"pipeline, placement_strategy, optimize, trace_device",
146140
[

0 commit comments

Comments
 (0)