[main] Add spatialdata element to segger export

pixi.toml

@@ -37,9 +37,11 @@ cudf-cu12 = "==24.10.*"
 cuml-cu12 = "==24.10.*"
 cugraph-cu12 = "==24.10.*"
 cupy-cuda12x = ">=12.2,<13.0"
+spatialdata = ">=0.4.0, <0.5"
+setuptools = ">=81.0.0, <82"
 
 [tool.uv.extra-build-dependencies]
 torch-scatter = ["torch"]
 
 [environments]
-cuda121 = ["cuda121"]
+cuda121 = ["cuda121"]

src/segger/__init__.py

@@ -1,6 +1,11 @@
 import logging
 import os
 from pathlib import Path
+
+import dask
+
+dask.config.set({"dataframe.query-planning": False})  # spatialdata doesn't yet support dask-expr; must be set before cudf pulls in dask.dataframe
+
 import cupy as cp
 import torch
 import rmm