MDAS: silence matplotlib deprecation warning

microsoft · Jan 27, 2025 · bb96d10 · bb96d10
1 parent 8a3b60d
commit bb96d10
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torchgeo/datasets/mdas.py b/torchgeo/datasets/mdas.py
@@ -7,11 +7,11 @@
 from collections.abc import Callable
 from typing import Any, ClassVar
 
-import matplotlib.cm as cm
 import matplotlib.pyplot as plt
 import numpy as np
 import rasterio as rio
 import torch
+from matplotlib.colormaps import get_cmap
 from matplotlib.colors import ListedColormap
 from matplotlib.figure import Figure
 from torch import Tensor
@@ -356,7 +356,7 @@ def plot(
                     axs[idx].imshow(img)
                 case 'osm_landuse_mask':
                     img = data.numpy().squeeze(0)
-                    cmap = ListedColormap([cm.get_cmap('tab20')(i) for i in range(20)])
+                    cmap = ListedColormap([get_cmap('tab20')(i) for i in range(20)])
                     im = axs[idx].imshow(img, cmap=cmap)
                     cbar = plt.colorbar(im, ax=axs[idx], ticks=range(19))
                     cbar.ax.set_yticklabels(