Skip to content

Commit

Permalink
Fix pandas DtypeWarning in download_demo (#2006)
Browse files Browse the repository at this point in the history
  • Loading branch information
fealho committed May 15, 2024
1 parent 08a464c commit 2d2b030
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion sdv/datasets/demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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]
Expand Down

0 comments on commit 2d2b030

Please sign in to comment.