Add COZI to nhood_enrichment()

pyproject.toml

@@ -56,6 +56,7 @@ dependencies = [
   "matplotlib-scalebar>=0.8",
   "networkx>=2.6",
   "numba>=0.56.4",
+  "numba-progress>=1.1",
   "numpy>=1.23",
   "omnipath>=1.0.7",
   "pandas>=2.1",
@@ -66,7 +67,7 @@ dependencies = [
   "scikit-image>=0.25",
   # due to https://github.com/scikit-image/scikit-image/issues/6850 breaks rescale ufunc
   "scikit-learn>=0.24",
-  "spatialdata>=0.7.2",  # 0.7.2 dropped xarray-schema (pkg_resources break, #1115)
+  "spatialdata>=0.7.2",          # 0.7.2 dropped xarray-schema (pkg_resources break, #1115)
   "spatialdata-plot>=0.3.3",
   "statsmodels>=0.12",
   # https://github.com/scverse/squidpy/issues/526

src/squidpy/datasets/__init__.py

@@ -6,6 +6,7 @@
     ImageDatasets,
     SpatialDataDatasets,
     VisiumDatasets,
+    cells,
     # AnnData datasets
     four_i,
     imc,
@@ -25,7 +26,6 @@
     visium_hne_image,
     visium_hne_image_crop,
     visium_hne_sdata,
-    cells,
 )
 
 __all__ = [