diff --git a/reader/import_data.py b/reader/import_data.py index 16e13cc51896fbd94b0a8f39483f4d4f9c881c8f..b298e1993225f9bf6f2e9e0eaae1617f8843efdd 100644 --- a/reader/import_data.py +++ b/reader/import_data.py @@ -61,10 +61,10 @@ def import_data(source,chunksize = None, skiprows = None): if isinstance(source,pd.io.parsers.TextFileReader): return source elif isinstance(source, io.StringIO): - TextParser = to_iterable_df(source,skiprows = None, chunksize = chunksize) + TextParser = to_iterable_df(source,skiprows = skiprows, chunksize = chunksize) return TextParser elif os.path.isfile(source): - TextParser = to_iterable_df(source,skiprows = None, chunksize = chunksize) + TextParser = to_iterable_df(source,skiprows = skiprows, chunksize = chunksize) return TextParser else: print('Error')