From 2d2b030d02a134d9648f03ac1c3f176e64a7a861 Mon Sep 17 00:00:00 2001 From: Felipe Alex Hofmann Date: Wed, 15 May 2024 10:09:02 -0700 Subject: [PATCH] Fix pandas DtypeWarning in download_demo (#2006) --- sdv/datasets/demo.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sdv/datasets/demo.py b/sdv/datasets/demo.py index f668ce403..43c35f4ea 100644 --- a/sdv/datasets/demo.py +++ b/sdv/datasets/demo.py @@ -96,7 +96,7 @@ def _get_data(modality, output_folder_name, in_memory_directory): for filename, file_ in in_memory_directory.items(): if filename.endswith('.csv'): table_name = Path(filename).stem - data[table_name] = pd.read_csv(io.StringIO(file_.decode())) + data[table_name] = pd.read_csv(io.StringIO(file_.decode()), low_memory=False) if modality != 'multi_table': data = data.popitem()[1]