docs: def save(path, im):
      """
      Saves an image to file.
  
      If the image is type float, it will assume to have values in [0, 1].
  
      Parameters
      ----------
      path : str
          Path to which the image will be saved.
      im : ndarray (image)
          Image.
      """
      from PIL import Image
      if im.dtype == np.uint8:
          pil_im = Image.fromarray(im)
      else:
          pil_im = Image.fromarray((im*255).astype(np.uint8))
      pil_im.save(path)