bug fixes for draw_property_layers (#2548)

projectmesa · Dec 14, 2024 · f72a268 · f72a268
1 parent 965eede
commit f72a268
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 9 deletions.
diff --git a/mesa/examples/advanced/sugarscape_g1mt/app.py b/mesa/examples/advanced/sugarscape_g1mt/app.py
@@ -1,9 +1,3 @@
-import os
-import sys
-
-sys.path.insert(0, os.path.abspath("../../../.."))
-
-
 import numpy as np
 import solara
 from matplotlib.figure import Figure

diff --git a/mesa/visualization/mpl_space_drawing.py b/mesa/visualization/mpl_space_drawing.py
@@ -182,7 +182,10 @@ def draw_property_layers(
 
     for layer_name, portrayal in propertylayer_portrayal.items():
         layer = property_layers.get(layer_name, None)
-        if not isinstance(layer, PropertyLayer):
+        if not isinstance(
+            layer,
+            PropertyLayer | mesa.experimental.cell_space.property_layer.PropertyLayer,
+        ):
             continue
 
         data = layer.data.astype(float) if layer.data.dtype == bool else layer.data
@@ -212,7 +215,7 @@ def draw_property_layers(
                 layer_name, [(0, 0, 0, 0), (*rgba_color[:3], alpha)]
             )
             im = ax.imshow(
-                rgba_data.transpose(1, 0, 2),
+                rgba_data,
                 origin="lower",
             )
             if colorbar:
@@ -226,7 +229,7 @@ def draw_property_layers(
             if isinstance(cmap, list):
                 cmap = LinearSegmentedColormap.from_list(layer_name, cmap)
             im = ax.imshow(
-                data.T,
+                data,
                 cmap=cmap,
                 alpha=alpha,
                 vmin=vmin,