diff --git a/error_generator/strategies/utilities/input_output.py b/error_generator/strategies/utilities/input_output.py index a6f19c7..0a89696 100644 --- a/error_generator/strategies/utilities/input_output.py +++ b/error_generator/strategies/utilities/input_output.py @@ -16,7 +16,8 @@ def read_csv_dataset(dataset_path, header_exists=True): keep_default_na=False, low_memory=False) dataset_dataframe = dataset_dataframe.apply(lambda x: x.str.strip()) - return [dataset_dataframe.columns.get_values().tolist()] + dataset_dataframe.get_values().tolist(),dataset_dataframe_version + return [dataset_dataframe.columns.to_list()] + dataset_dataframe.to_numpy().tolist(), dataset_dataframe_version + else: dataset_dataframe = pandas.read_csv(dataset_path, sep=",", header=None, encoding="utf-8", dtype=str, keep_default_na=False)