502d6ebfcd698061cbdc1b369cb7ff449ff9db08,tests/unit/test_estimator.py,,test_init_with_source_dir_s3,#Any#Any#,330

Before Change

                        sagemaker_session=sagemaker_session,
                        train_instance_count=INSTANCE_COUNT, train_instance_type=INSTANCE_TYPE,
                        enable_cloudwatch_metrics=False)
    fw.fit("s3://{}".format(uri))

    expected_hyperparameters = BASE_HP.copy()
    expected_hyperparameters["sagemaker_enable_cloudwatch_metrics"] = "false"
    expected_hyperparameters["sagemaker_container_log_level"] = str(logging.INFO)

After Change

                        enable_cloudwatch_metrics=False)
    fw._prepare_for_training()

    expected_hyperparameters = {
        "sagemaker_program": SCRIPT_NAME,
        "sagemaker_job_name": JOB_NAME,
        "sagemaker_enable_cloudwatch_metrics": False,
        "sagemaker_container_log_level": logging.INFO,
        "sagemaker_submit_directory": "s3://location",
        "sagemaker_region": "us-west-2",
    }
    assert fw._hyperparameters == expected_hyperparameters


// _TrainingJob "utils"

Instances