persistent_block - collect func

Author: Canicula98

src/dataclay/contrib/persistent_block.py

@@ -11,7 +11,8 @@
     from pycompss.api.parameter import IN
 except ImportError:
     from dataclay.contrib.pycompss_dummy.dummy_pycompss import task, IN
-
+import logging
+logger = logging.getLogger(__name__)
 
 class PersistentBlock(DataClayObject):
     block_data: np.ndarray
@@ -77,6 +78,11 @@ def partial_histogram(self, n_bins: int, n_dimensions: int) -> np.ndarray:
         values, _ = np.histogramdd(self.block_data, n_bins, [(0, 1)] * n_dimensions)
         return values
 
+    @activemethod
+    def collect(self, squeeze=True):
+        return self.block_data
+
     @activemethod
     def block_apply(self, func, *args, **kwargs):
-        return func(self.block_data, *args, **kwargs)
+        self.block_data=func(self.block_data,*args, **kwargs)
+        return self.block_data