 Display generated image as tiles wave = pipeline_output.waves[0] tile_result = wave.argmax(axis=2)  from wfd.scmap import demo_map_image display(demo_map_image(tile_result, wfc_data_path = wfc_data_path))  