Skip to content

Commit

Permalink
make sure we optimize the limits in eager readers
Browse files Browse the repository at this point in the history
  • Loading branch information
ritchie46 committed Jan 5, 2022
1 parent 73e5a33 commit 1d9dd25
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions py-polars/polars/internals/frame.py
Original file line number Diff line number Diff line change
Expand Up @@ -512,7 +512,7 @@ def read_csv(
rechunk=rechunk,
)
if columns is None:
return scan.collect(no_optimization=True)
return scan.collect()
elif is_str_sequence(columns, False):
return scan.select(columns).collect()
else:
Expand Down Expand Up @@ -574,7 +574,7 @@ def read_parquet(
scan = scan_parquet(file, n_rows=n_rows, rechunk=True, parallel=parallel)

if columns is None:
return scan.collect(no_optimization=True)
return scan.collect()
elif is_str_sequence(columns, False):
return scan.select(columns).collect()
else:
Expand Down

0 comments on commit 1d9dd25

Please sign in to comment.