fix get_cmap import (#392)

* fix get_cmap import * Update structuregraph.py
materialsproject · May 23, 2024 · dcd3c2f · dcd3c2f
1 parent 1aa9b00
commit dcd3c2f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/crystal_toolkit/core/legend.py b/crystal_toolkit/core/legend.py
@@ -6,7 +6,7 @@
 from itertools import chain
 from typing import TYPE_CHECKING, Any
 
-from matplotlib.cm import get_cmap
+from matplotlib.pyplot import get_cmap
 from monty.json import MSONable
 from monty.serialization import loadfn
 from palettable.colorbrewer.qualitative import Set1_9

diff --git a/crystal_toolkit/renderables/structuregraph.py b/crystal_toolkit/renderables/structuregraph.py
@@ -5,7 +5,7 @@
 from typing import Sequence
 
 import numpy as np
-from matplotlib.cm import get_cmap
+from matplotlib.pyplot import get_cmap
 from pymatgen.analysis.graphs import StructureGraph
 from pymatgen.core import PeriodicSite