code: def plot_depth(
      depth, ax=None, depth_range=None, colormap="inferno_r"
  ):
      if ax is None:
          _, ax = get_image_plot(depth.shape[1], depth.shape[0])
      if depth_range is None:
          depth_range = np.min(depth), np.max(depth)
      ax.imshow(
          (depth - depth_range[0]) / (depth_range[1] - depth_range[0]),
          cmap=colormap,
          interpolation="nearest",
      )
  
file_path: kubric/kubric/plotting.py
imports: import colorsys
  import numpy
label_desc: Use image node_id 4 for calling a global function with example usage: plot_depth(depth, ax, depth_range, colormap) without return types