feat: Removes matplotlib warnings

frgfm · Oct 19, 2023 · 98621d6 · 98621d6
1 parent 72d9e33
commit 98621d6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/pyproject.toml b/pyproject.toml
@@ -37,7 +37,7 @@ dependencies = [
  # cf. https://github.com/pytorch/vision/issues/4934
  # https://github.com/frgfm/Holocron/security/dependabot/5
  "Pillow>=8.4.0,!=9.2.0",
- "matplotlib>=3.0.0,<4.0.0",
+ "matplotlib>=3.7.0,<4.0.0",
 ]
 
 [project.optional-dependencies]

diff --git a/torchcam/utils.py b/torchcam/utils.py
@@ -4,7 +4,7 @@
 # See LICENSE or go to <https://www.apache.org/licenses/LICENSE-2.0> for full license details.
 
 import numpy as np
-from matplotlib import cm
+from matplotlib import colormaps as cm
 from PIL import Image