Fixed bug passing default None skiprows

6cf10acf · iregon · a7bc9c12 · 6cf10acf
Commit 6cf10acf authored 5 years ago by iregon
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

reader/import_data.py reader/import_data.py +2 -2

No files found.
--- 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')