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)