Update functions/heat_map_functions.py

functions/heat_map_functions.py

@@ -535,7 +535,10 @@ def heat_map_plot_hex_damage(df:pl.DataFrame,
                     x='px',
                     y='pz',
                     ec = 'black',
-                    cmap=cmap,
+                    palette=cmap,
+                    hue='woba_pred_contact',
+                    hue_norm=mcolors.Normalize(vmin=0, vmax=0.5),  # <- HERE
+                    legend=False
                     ax=ax)