grok/visualization.py [326:331]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        ax.legend()
    else:
        fig.colorbar(
            sm,
            ax=ax,
            label="% training data",
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



scripts/visualize_metrics.py [184:189]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
        ax.legend()
    else:
        fig.colorbar(
            sm,
            ax=ax,
            label="% training data",
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -