def METHOD_NAME():
    """Tests 2d regression normalization."""
    y = np.random.rand(10, 5)
    y_expected = y
    y_out = normalize_prediction_shape(y, mode="regression", n_tasks=5)
    assert y_out.shape == (10, 5)
    assert np.array_equal(y_expected, y_out)