fix pylint check

adap · Aug 27, 2024 · 5a25e5e · 5a25e5e
1 parent 1679879
commit 5a25e5e
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/datasets/flwr_datasets/partitioner/image_semantic_partitioner.py b/datasets/flwr_datasets/partitioner/image_semantic_partitioner.py
@@ -391,9 +391,7 @@ def _determine_partition_id_to_indices_if_needed(self) -> None:
 
     def _preprocess_dataset_images(self, indices: List[int]) -> NDArrayFloat:
         """Preprocess the images in the dataset."""
-        images = np.array(
-            self.dataset[indices][self._image_column_name], dtype=np.float32
-        )
+        images = np.array(self.dataset[indices][self._image_column_name], dtype=float)
         if len(images.shape) == 3:  # [B, H, W]
             images = np.reshape(
                 images, (images.shape[0], 1, images.shape[1], images.shape[2])