feat(datasets) Improve error message for incorrect dataset type (#4114)

adap · Sep 13, 2024 · 1c894d2 · 1c894d2
1 parent 541064b
commit 1c894d2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py
@@ -317,7 +317,8 @@ def _prepare_dataset(self) -> None:
             raise ValueError(
                 "Probably one of the specified parameter in `load_dataset_kwargs` "
                 "change the return type of the datasets.load_dataset function. "
-                "Make sure to use parameter such that the return type is DatasetDict."
+                "Make sure to use parameter such that the return type is DatasetDict. "
+                f"The return type is currently: {type(self._dataset)}."
             )
         if self._shuffle:
             # Note it shuffles all the splits. The self._dataset is DatasetDict