diff --git a/src/sagemaker/fw_utils.py b/src/sagemaker/fw_utils.py index fb6491fb68..4c97032384 100644 --- a/src/sagemaker/fw_utils.py +++ b/src/sagemaker/fw_utils.py @@ -327,9 +327,6 @@ def validate_mp_config(config): ValueError: If any of the keys have incorrect values. """ - if "partitions" not in config: - raise ValueError("'partitions' is a required parameter.") - def validate_positive(key): try: if not isinstance(config[key], int) or config[key] < 1: diff --git a/tests/unit/test_fw_utils.py b/tests/unit/test_fw_utils.py index c80daeab3d..b4fb76a798 100644 --- a/tests/unit/test_fw_utils.py +++ b/tests/unit/test_fw_utils.py @@ -135,12 +135,6 @@ def test_tar_and_upload_dir_s3_without_kms_with_overridden_settings(utils, sagem obj.upload_file.assert_called_with(utils.create_tar_file(), ExtraArgs=None) -def test_mp_config_partition_exists(): - mp_parameters = {} - with pytest.raises(ValueError): - fw_utils.validate_mp_config(mp_parameters) - - @pytest.mark.parametrize( "pipeline, placement_strategy, optimize, trace_device", [