User Prompt

Training examples (input -> output): {training_examples_text}

Shapes (rows x cols): {shape_summary}

Test inputs: {test_inputs_text}

Respond with ONE python code block (python ...), defining: def transform(grid: np.ndarray) -> np.ndarray