diff --git a/CHANGELOG.rst b/CHANGELOG.rst index 2c04fe4e85..599f79e3cc 100644 --- a/CHANGELOG.rst +++ b/CHANGELOG.rst @@ -6,6 +6,7 @@ CHANGELOG ======== * bug-fix: Local Mode: Create output/data directory expected by SageMaker Container. +* bug-fix: Estimator accepts the vpc configs made capable by 1.9.1 1.9.2 ===== diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py index 9febff5240..0c9a10afed 100644 --- a/src/sagemaker/estimator.py +++ b/src/sagemaker/estimator.py @@ -429,7 +429,7 @@ class Estimator(EstimatorBase): def __init__(self, image_name, role, train_instance_count, train_instance_type, train_volume_size=30, train_max_run=24 * 60 * 60, input_mode='File', output_path=None, output_kms_key=None, base_job_name=None, sagemaker_session=None, - hyperparameters=None): + hyperparameters=None, tags=None, subnets=None, security_group_ids=None): """Initialize an ``Estimator`` instance. Args: @@ -467,7 +467,8 @@ def __init__(self, image_name, role, train_instance_count, train_instance_type, self.hyperparam_dict = hyperparameters.copy() if hyperparameters else {} super(Estimator, self).__init__(role, train_instance_count, train_instance_type, train_volume_size, train_max_run, input_mode, - output_path, output_kms_key, base_job_name, sagemaker_session) + output_path, output_kms_key, base_job_name, sagemaker_session, + tags, subnets, security_group_ids) def train_image(self): """