TypeError if custom dataset is not DatasetDict

adap · Apr 4, 2024 · 4c67721 · 4c67721
1 parent 5b1e523
commit 4c67721
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/datasets/flwr_datasets/utils.py b/datasets/flwr_datasets/utils.py
@@ -87,9 +87,8 @@ def _check_if_dataset_tested(dataset: Union[str, DatasetDict]) -> None:
             stacklevel=1,
         )
     if type(dataset) is not DatasetDict:
-        warnings.warn(
-            f"The provided dataset is not in the DatasetDict format.",
-            stacklevel=1,
+        raise TypeError(
+            "The provided dataset is not in the DatasetDict format."
         )