Add UltraLayout: Advanced constraint-based positioning for non-orthogonal subplot arrangements

ultraplot/axes/plot.py

@@ -3287,9 +3287,9 @@ def _parse_level_lim(
         for z in zs:
             if z is None:  # e.g. empty scatter color
                 continue
+            z = inputs._to_numpy_array(z)
             if z.ndim > 2:  # e.g. imshow data
                 continue
-            z = inputs._to_numpy_array(z)
             if inbounds and x is not None and y is not None:  # ignore if None coords
                 z = self._inbounds_vlim(x, y, z, to_centers=to_centers)
             imin, imax = inputs._safe_range(z, pmin, pmax)

ultraplot/figure.py

@@ -1841,6 +1841,8 @@ def _axes_dict(naxs, input, kw=False, default=None):
         # Create or update the gridspec and add subplots with subplotspecs
         # NOTE: The gridspec is added to the figure when we pass the subplotspec
         if gs is None:
+            if "layout_array" not in gridspec_kw:
+                gridspec_kw = {**gridspec_kw, "layout_array": array}
             gs = pgridspec.GridSpec(*array.shape, **gridspec_kw)
         else:
             gs.update(**gridspec_kw)