Fixing dataset name check: if neither DatasetDict or str

adap · Apr 4, 2024 · e5fb7e1 · e5fb7e1
1 parent 26ed66d
commit e5fb7e1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/datasets/flwr_datasets/utils.py b/datasets/flwr_datasets/utils.py
@@ -86,7 +86,7 @@ def _check_if_dataset_tested(dataset: Union[str, DatasetDict]) -> None:
             f"The currently tested dataset are {tested_datasets}. Given: {dataset}.",
             stacklevel=1,
         )
-    if type(dataset) is not DatasetDict:
+    elif type(dataset) not in [DatasetDict, str]:
         raise TypeError(
             "The provided dataset is not in the DatasetDict format."
         )