From 19a549e105e8a4aa61ede41a50b94bdc7ae1e39a Mon Sep 17 00:00:00 2001 From: Merel Theisen <49397448+merelcht@users.noreply.github.com> Date: Mon, 27 Nov 2023 11:30:48 +0000 Subject: [PATCH] chore(datasets): Fix `pandas.GenericDataset` doctest (#445) Fix pandas.GenericDataset doctest Signed-off-by: Merel Theisen --- Makefile | 1 - kedro-datasets/kedro_datasets/pandas/generic_dataset.py | 4 +++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/Makefile b/Makefile index da2ac100b..5ed915a8b 100644 --- a/Makefile +++ b/Makefile @@ -31,7 +31,6 @@ dataset-doctests: --ignore kedro_datasets/matplotlib/matplotlib_writer.py \ --ignore kedro_datasets/pandas/deltatable_dataset.py \ --ignore kedro_datasets/pandas/gbq_dataset.py \ - --ignore kedro_datasets/pandas/generic_dataset.py \ --ignore kedro_datasets/pandas/sql_dataset.py \ --ignore kedro_datasets/partitions/incremental_dataset.py \ --ignore kedro_datasets/partitions/partitioned_dataset.py \ diff --git a/kedro-datasets/kedro_datasets/pandas/generic_dataset.py b/kedro-datasets/kedro_datasets/pandas/generic_dataset.py index 96bf7d253..b48e0f9ba 100644 --- a/kedro-datasets/kedro_datasets/pandas/generic_dataset.py +++ b/kedro-datasets/kedro_datasets/pandas/generic_dataset.py @@ -70,7 +70,9 @@ class GenericDataset(AbstractVersionedDataset[pd.DataFrame, pd.DataFrame]): >>> >>> data = pd.DataFrame({"col1": [1, 2], "col2": [4, 5], "col3": [5, 6]}) >>> - >>> dataset = GenericDataset(filepath="test.csv", file_format="csv") + >>> dataset = GenericDataset( + ... filepath="test.csv", file_format="csv", save_args={"index": False} + ... ) >>> dataset.save(data) >>> reloaded = dataset.load() >>> assert data.equals(reloaded)