visualization/__init__.py

from .overlay import (
    overlay_heatmap,
    apply_colormap,
    make_comparison_figure,
    make_plotly_heatmap,
    plot_top_predictions,
    compute_attribution_stats,
    fig_to_pil,
    COLORMAPS,
)

__all__ = [
    "overlay_heatmap", "apply_colormap",
    "make_comparison_figure", "make_plotly_heatmap",
    "plot_top_predictions", "compute_attribution_stats",
    "fig_to_pil", "COLORMAPS",
]