Skip to content

GitLab

Commit 6cf10acf authored by iregon's avatar iregon
Browse files

Fixed bug passing default None skiprows

parent a7bc9c12
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
reader/import_data.py
View file @ 6cf10acf
...@@ -61,10 +61,10 @@ def import_data(source,chunksize = None, skiprows = None): ...@@ -61,10 +61,10 @@ def import_data(source,chunksize = None, skiprows = None):
if isinstance(source,pd.io.parsers.TextFileReader): if isinstance(source,pd.io.parsers.TextFileReader):
return source return source
elif isinstance(source, io.StringIO): 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 return TextParser
elif os.path.isfile(source): 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 return TextParser
else: else:
print('Error') print('Error')
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment