diff --git a/datasets/doc/source/how-to-use-with-local-data.rst b/datasets/doc/source/how-to-use-with-local-data.rst index a56603809e40..276f6d6936ee 100644 --- a/datasets/doc/source/how-to-use-with-local-data.rst +++ b/datasets/doc/source/how-to-use-with-local-data.rst @@ -173,6 +173,7 @@ From dictionary From list ^^^^^^^^^ .. code-block:: python + from datasets import Dataset from flwr_datasets.partitioner import ChosenPartitioner @@ -190,6 +191,7 @@ From list From pd.DataFrame ^^^^^^^^^^^^^^^^^ .. code-block:: python + from datasets import Dataset from flwr_datasets.partitioner import ChosenPartitioner