fix: Fixing compatibility with merged changes.

Author: Thomas Zilio

zcollection/collection/__init__.py

@@ -635,12 +635,12 @@ def update(
             storage.execute_transaction(
                 client, self.synchronizer,
                 client.map(local_func,
-                       tuple(batches),
-                       key=func.__name__,
-                       func_args=args,
-                       func_kwargs=kwargs))
+                           tuple(batches),
+                           key=func.__name__,
+                           func_args=args,
+                           func_kwargs=kwargs))
         else:
-            local_func(selected_partitions)
+            local_func(selected_partitions, args, kwargs)
         tuple(map(self.fs.invalidate_cache, selected_partitions))
 
     def drop_variable(

zcollection/collection/callable_objects.py

@@ -16,8 +16,8 @@
 
 #: Function type to load and call a callback function of type
 #: :class:`PartitionCallable`.
-WrappedPartitionCallable = Callable[[Sequence[str], list[Any], dict[str, Any]],
-                                    None]
+WrappedPartitionCallable = Callable[
+    [Sequence[str], tuple[Any, ...], dict[str, Any]], None]
 
 
 #: pylint: disable=too-few-public-methods

zcollection/collection/detail.py

@@ -276,7 +276,7 @@ def _wrap_update_func(
         array.
     """
 
-    def wrap_function(partitions: Iterable[str], func_args: list[Any],
+    def wrap_function(partitions: Iterable[str], func_args: tuple[Any, ...],
                       func_kwargs: dict[str, Any]) -> None:
         # Applying function for each partition's data
         for partition in partitions:
@@ -328,7 +328,7 @@ def _wrap_update_func_with_overlap(
     if depth < 0:
         raise ValueError('Depth must be non-negative.')
 
-    def wrap_function(partitions: Sequence[str], func_args: list[Any],
+    def wrap_function(partitions: Sequence[str], func_args: tuple[Any, ...],
                       func_kwargs: dict[str, Any]) -> None:
         # Applying function for each partition's data
         for partition in partitions:

zcollection/view/__init__.py

@@ -610,15 +610,17 @@ def update(
             batches: Iterator[Sequence[Any]] = dask_utils.split_sequence(
                 datasets_list, npartitions
                 or dask_utils.dask_workers(client, cores_only=True))
+
             awaitables: list[dask.distributed.Future] = client.map(
                 wrap_function,
                 tuple(batches),
                 key=func.__name__,
                 base_dir=self.base_dir,
-            func_args=args,
-            func_kwargs=kwargs)storage.execute_transaction(client, self.synchronizer, awaitables)
+                func_args=args,
+                func_kwargs=kwargs)
+            storage.execute_transaction(client, self.synchronizer, awaitables)
         else:
-            wrap_function(datasets_list, self.base_dir)
+            wrap_function(datasets_list, self.base_dir, args, kwargs)
 
     # pylint: disable=duplicate-code
     # false positive, no code duplication

zcollection/view/detail.py

@@ -38,9 +38,9 @@
 from ..type_hints import ArrayLike, NDArray
 
 #: Type of the function used to update a view.
-ViewUpdateCallable = Callable[
-    [Iterable[tuple[dataset.Dataset, str]], str, list[Any], dict[str,
-                                                                 Any]], None]
+ViewUpdateCallable = Callable[[
+    Iterable[tuple[dataset.Dataset, str]], str, tuple[Any, ...], dict[str, Any]
+], None]
 
 #: Name of the file that contains the checksum of the view.
 CHECKSUM_FILE = '.checksum'
@@ -383,7 +383,7 @@ def _wrap_update_func(
     """
 
     def wrap_function(parameters: Iterable[tuple[dataset.Dataset, str]],
-                      base_dir: str, func_args: list[Any],
+                      base_dir: str, func_args: tuple[Any, ...],
                       func_kwargs: dict[str, Any]) -> None:
         """Wrap the function to be applied to the dataset."""
         for zds, partition in parameters:
@@ -428,7 +428,7 @@ def _wrap_update_func_overlap(
         raise ValueError('The depth must be positive')
 
     def wrap_function(parameters: Iterable[tuple[dataset.Dataset, str]],
-                      base_dir: str, func_args: list[Any],
+                      base_dir: str, func_args: tuple[Any, ...],
                       func_kwargs: dict[str, Any]) -> None:
         """Wrap the function to be applied to the dataset."""
         zds: dataset.Dataset