[FIX] Fix import error due to deprecated matplotlib function

[eric2302]

Matplotlib deprecated register_cmap function since 3.7.0, so an equivalent function is now being called under the same name