public-mirrors/3b1b-manim
1
0
Fork 0
mirror of https://github.com/3b1b/manim.git synced 2025-04-13 09:47:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix import of pyplot

Browse source
This commit is contained in:
Grant Sanderson 2024-11-25 10:49:05 -07:00
parent 1d0deb8a33
commit ed2f9f3305
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
manimlib/utils/color.py
View file

@ -5,6 +5,7 @@ from colour import hex2rgb
from colour import rgb2hex from colour import rgb2hex
import numpy as np import numpy as np
import random import random
from matplotlib import pyplot
from manimlib.constants import COLORMAP_3B1B from manimlib.constants import COLORMAP_3B1B
from manimlib.constants import WHITE from manimlib.constants import WHITE
@ -157,7 +158,7 @@ def get_colormap_from_colors(colors: Iterable[ManimColor]) -> Callable[[Sequence
def get_color_map(map_name: str) -> Callable[[Sequence[float]], Vect4Array]: def get_color_map(map_name: str) -> Callable[[Sequence[float]], Vect4Array]:
if map_name == "3b1b_colormap": if map_name == "3b1b_colormap":
return get_colormap_from_colors(COLORMAP_3B1B) return get_colormap_from_colors(COLORMAP_3B1B)
return plt.get_cmap(map_name) return pyplot.get_cmap(map_name)
# Delete this? # Delete this?