From 915dc3574239aa6ec19e8fa9e2fc26e2f6260317 Mon Sep 17 00:00:00 2001
From: s214735 <s214735@dtu.dk>
Date: Mon, 30 Dec 2024 11:35:52 +0100
Subject: [PATCH] dask fixes

---
 qim3d/io/_ome_zarr.py | 4 ++--
 qim3d/io/_saving.py   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/qim3d/io/_ome_zarr.py b/qim3d/io/_ome_zarr.py
index 0cf4803e..983aab01 100644
--- a/qim3d/io/_ome_zarr.py
+++ b/qim3d/io/_ome_zarr.py
@@ -70,10 +70,10 @@ class OMEScaler(
         This method performs multi-scale downsampling on a 3D dataset, generating image pyramids. It processes the data in chunks using Dask.
 
         Args:
-            base (dask.array): The 3D array (volume) to be downsampled. Must be a Dask array for chunked processing.
+            base (dask.array.core.array): The 3D array (volume) to be downsampled. Must be a Dask array for chunked processing.
 
         Returns:
-            list of dask.array: A list of downsampled volumes, where each element represents a different scale. The first element corresponds to the original resolution, and subsequent elements represent progressively downsampled versions.
+            list of dask.array.core.Array: A list of downsampled volumes, where each element represents a different scale. The first element corresponds to the original resolution, and subsequent elements represent progressively downsampled versions.
 
         The downsampling process occurs scale by scale, using the following steps:
         - For each scale, the array is resized based on the downscale factor, computed as a function of the current scale level.
diff --git a/qim3d/io/_saving.py b/qim3d/io/_saving.py
index 830674a2..ff713a4f 100644
--- a/qim3d/io/_saving.py
+++ b/qim3d/io/_saving.py
@@ -260,7 +260,7 @@ class DataSaver:
 
         Args:
             path (str): The path to the Zarr array on disk.
-            data (dask.array): The Dask array to be saved to disk.
+            data (dask.array.core.Array): The Dask array to be saved to disk.
 
         Returns:
             zarr.core.Array: The Zarr array saved on disk.
-- 
GitLab