Merge pull request #485 from aai-institute/bugfix/484-type-error-dask

Bug in handling single dimensional arrays in DaskInfluenceCalculator

CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## Unreleased
+
+### Fixed
+
+- Bug in using `DaskInfluenceCalcualator` with `TorchnumpyConverter`
+  for single dimensional arrays [PR #485](https://github.com/aai-institute/pyDVL/pull/485)
+
 ## 0.8.0 - 🆕 New interfaces, scaling computation, bug fixes and improvements 🎁
 
 ### Added

src/pydvl/influence/influence_calculator.py

@@ -288,8 +288,8 @@ def func(x_numpy: NDArray, y_numpy: NDArray, model: InfluenceFunctionModel):
             chunk_shape = (chunk_size, self.n_parameters)
             chunk_array = da.from_delayed(
                 delayed(func)(
-                    x_chunk.squeeze().tolist(),
-                    y_chunk.squeeze().tolist(),
+                    x_chunk.squeeze()[()],
+                    y_chunk.squeeze()[()],
                     self.influence_function_model,
                 ),
                 dtype=x.dtype,
@@ -400,10 +400,10 @@ def func(
 
                 block_array = da.from_delayed(
                     delayed(func)(
-                        x_test_chunk.squeeze().tolist(),
-                        y_test_chunk.squeeze().tolist(),
-                        x_chunk.squeeze().tolist(),
-                        y_chunk.squeeze().tolist(),
+                        x_test_chunk.squeeze()[()],
+                        y_test_chunk.squeeze()[()],
+                        x_chunk.squeeze()[()],
+                        y_chunk.squeeze()[()],
                         self.influence_function_model,
                     ),
                     shape=block_shape,
@@ -506,9 +506,9 @@ def func(
 
                 block_array = da.from_delayed(
                     delayed(func)(
-                        z_test_chunk.squeeze().tolist(),
-                        x_chunk.squeeze().tolist(),
-                        y_chunk.squeeze().tolist(),
+                        z_test_chunk.squeeze()[()],
+                        x_chunk.squeeze()[()],
+                        y_chunk.squeeze()[()],
                         self.influence_function_model,
                     ),
                     shape=block_shape,