diff --git a/pyproject.toml b/pyproject.toml index b929f91..21d2bec 100644 --- 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 index 877b66a..8428baa 100644 --- a/torchcam/utils.py +++ b/torchcam/utils.py @@ -4,7 +4,7 @@ # See LICENSE or go to for full license details. import numpy as np -from matplotlib import cm +from matplotlib import colormaps as cm from PIL import Image