From e7661940f4e761cbe2b61b2c570f33c2dcdb09bb Mon Sep 17 00:00:00 2001
From: Luca Marconato <m.lucalmer@gmail.com>
Date: Wed, 6 Aug 2025 12:28:45 +0200
Subject: [PATCH] z support get_dims for shapes

---
 src/spatialdata/models/_utils.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/src/spatialdata/models/_utils.py b/src/spatialdata/models/_utils.py
index eeaa7ecd9..595fd9ae5 100644
--- a/src/spatialdata/models/_utils.py
+++ b/src/spatialdata/models/_utils.py
@@ -164,6 +164,8 @@ def _(e: GeoDataFrame) -> tuple[str, ...]:
     all_dims = (X, Y, Z)
     n = e.geometry.iloc[0]._ndim
     dims = all_dims[:n]
+    if Z not in dims and Z in e.columns:
+        dims += (Z,)
     _validate_dims(dims)
     return dims