Add Polygon and PatchCollection imports for visualization

projectmesa · Jan 10, 2025 · f17e99d · f17e99d
1 parent d7c570a
commit f17e99d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mesa/visualization/mpl_space_drawing.py b/mesa/visualization/mpl_space_drawing.py
@@ -12,12 +12,13 @@
 from collections.abc import Callable
 from typing import Any
 
+from matplotlib.patches import Polygon
 import networkx as nx
 import numpy as np
 from matplotlib import pyplot as plt
 from matplotlib.axes import Axes
 from matplotlib.cm import ScalarMappable
-from matplotlib.collections import LineCollection
+from matplotlib.collections import LineCollection, PatchCollection
 from matplotlib.colors import LinearSegmentedColormap, Normalize, to_rgba
 
 import mesa