code: def fill_image(rgb: Collection[int], image_size: int):
      """Test helper function to create images.
  
      Args:
        rgb: A tuple or array of rgb values in [r, g, b] format
        image_size: Int specifying the edge of the square image
  
      Returns:
        Numpy array of shape (image_size, image_size, 3) filled with the rgb color
      """
      r, g, b = rgb
      return np.broadcast_to(
          np.array([[[r, g, b]]], dtype=np.uint8),
          shape=(image_size, image_size, 3),
      )
  
imports: from typing import Collection
  import numpy
  import tensorflow
label_desc: Use image node_id 1 for calling a global function with example usage: fill_image(rgb, image_size) and returns: np