Skip to content

Commit

Permalink
polars updates
Browse files Browse the repository at this point in the history
mdancho84 committed Oct 25, 2023
1 parent f0e9ae0 commit 71823e1
Showing 1 changed file with 3 additions and 2 deletions.
5 changes: 3 additions & 2 deletions src/pytimetk/utils/pandas_helpers.py
Original file line number Diff line number Diff line change
@@ -1,5 +1,6 @@
import pandas as pd
import pandas_flavor as pf
import polars as pl

from pytimetk.utils.checks import check_dataframe_or_groupby

@@ -49,9 +50,9 @@ def glimpse(
check_dataframe_or_groupby(data)

if engine == 'pandas':
return _glimpse_polars(df, max_width)
return _glimpse_pandas(data, max_width)
elif engine == 'polars':
return _glimpse_polars(df, max_width)
return _glimpse_polars(data, max_width)
else:
raise ValueError("Invalid engine. Use 'pandas' or 'polars'.")

0 comments on commit 71823e1

Please sign in to comment.