a33f747920882f6cd857bbdb50050ea360a6f366,tests/integration_tests/functional_api_test.py,,test_image_blocks,#Any#,69

Before Change

def test_image_blocks(tmp_path):
    num_instances = 10
    (x_train, y_train), (x_test, y_test) = mnist.load_data()
    x_train = x_train[:num_instances]
    y_train = y_train[:num_instances]

After Change

def test_image_blocks(tmp_path):
    num_instances = 10
    x_train = utils.generate_data(num_instances=num_instances, shape=(28, 28))
    y_train = np.random.randint(0, 10, num_instances)

    input_node = ak.ImageInput()
    output = ak.Normalization()(input_node)
    output = ak.ImageAugmentation()(output)

Instances