def get_colors()

in utils.py [0:0]


def get_colors(num_samples = 3, end_point = 1.0):
  sample_points = np.linspace(0.1, end_point, num_samples)
  return plt.cm.viridis(sample_points)