Fix matplotlib deprecation

albermax · Sep 13, 2023 · 488a0fe · 488a0fe
1 parent 78acb3f
commit 488a0fe
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/innvestigate/utils/visualizations.py b/src/innvestigate/utils/visualizations.py
@@ -1,5 +1,6 @@
 from __future__ import annotations
 
+import matplotlib as mpl
 import matplotlib.pyplot as plt
 import numpy as np
 
@@ -57,7 +58,7 @@ def heatmap(
     :param kwargs: Arguments passed on to :func:`project`
     :return: The tensor as color-map.
     """
-    cmap = plt.cm.get_cmap(cmap_type)
+    cmap = mpl.colormaps.get_cmap(cmap_type)
 
     tmp = X
     shape = tmp.shape