miniworld.utils.output_visualize.visualize_2D_heatmap

miniworld.utils.output_visualize.visualize_2D_heatmap(heatmap, file_name='heatmap', heatmap_dir='opt_visualize/')