c21141762d380713b30ed4a0759e5628c49b095e,tests/integ/test_tuner.py,,kmeans_train_set,#Any#,56

Before Change

@pytest.fixture(scope="module")
def kmeans_train_set(sagemaker_session):
    data_path = os.path.join(DATA_DIR, "one_p_mnist", "mnist.pkl.gz")
    pickle_args = {} if sys.version_info.major == 2 else {"encoding": "latin1"}
    // Load the data into memory as numpy arrays
    with gzip.open(data_path, "rb") as f:
        train_set, _, _ = pickle.load(f, **pickle_args)

    return train_set


@pytest.fixture(scope="module")

After Change

@pytest.fixture(scope="module")
def kmeans_train_set(sagemaker_session):
    return datasets.one_p_mnist()


@pytest.fixture(scope="module")
def kmeans_estimator(sagemaker_session, cpu_instance_type):

Instances