Skip to content
Open
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
11 changes: 10 additions & 1 deletion artemis/plotting/pyplot_plus.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,16 @@
import logging
import numpy as np
import matplotlib.colors as colors
from matplotlib.cm import register_cmap
try:
from matplotlib.cm import register_cmap
except ImportError:
try:
from matplotlib.colors import register_cmap
except ImportError:
# For newer matplotlib versions (>=3.5.0)
from matplotlib import colormaps
def register_cmap(name, cmap):
colormaps.register(cmap=cmap, name=name)
from matplotlib.colors import LinearSegmentedColormap
from si_prefix import si_format
import matplotlib
Expand Down