Skip to content

Commit 5d66d76

Browse files
authored
fix: update AutoML default max_candidate value to use the service default (#1491)
1 parent 69e264f commit 5d66d76

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

src/sagemaker/automl/automl.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -39,7 +39,7 @@ def __init__(
3939
encrypt_inter_container_traffic=False,
4040
vpc_config=None,
4141
problem_type=None,
42-
max_candidates=500,
42+
max_candidates=None,
4343
max_runtime_per_training_job_in_seconds=None,
4444
total_job_runtime_in_seconds=None,
4545
job_objective=None,

tests/unit/sagemaker/automl/test_auto_ml.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@
3232
DEFAULT_S3_INPUT_DATA = "s3://{}/data".format(BUCKET_NAME)
3333
DEFAULT_OUTPUT_PATH = "s3://{}/".format(BUCKET_NAME)
3434
LOCAL_DATA_PATH = "file://data"
35-
DEFAULT_MAX_CANDIDATES = 500
35+
DEFAULT_MAX_CANDIDATES = None
3636
DEFAULT_JOB_NAME = "automl-{}".format(TIMESTAMP)
3737

3838
JOB_NAME = "default-job-name"

0 commit comments

Comments
 (0)