diff --git a/demo.py b/demo.py index 577b5396b..62ac4f46c 100644 --- a/demo.py +++ b/demo.py @@ -2,8 +2,8 @@ import polars as pl from great_tables.data import sp500 -from polars_api_compat import to_original_object -from polars_api_compat import to_polars_api +from puffin import to_original_object +from puffin import to_polars_api # Define the start and end dates for the data range start_date = "2010-06-07" diff --git a/f.py b/f.py index a66a60890..75cec5bbd 100644 --- a/f.py +++ b/f.py @@ -3,7 +3,7 @@ import pandas as pd import polars as pl -from polars_api_compat import to_polars_api, to_original_object +from puffin import to_polars_api, to_original_object AnyDataFrame = TypeVar("AnyDataFrame") diff --git a/tests/test_common.py b/tests/test_common.py index 8a2eaa398..0ea8f6ac9 100644 --- a/tests/test_common.py +++ b/tests/test_common.py @@ -1,12 +1,12 @@ # ruff: noqa import pandas as pd -import polars_api_compat +import puffin def test_common(): df = pd.DataFrame({"a": [1, 2, 3], "b": [4, 5, 6]}) - dfx, plx = polars_api_compat.to_polars_api(df, api_version="0.20") + dfx, plx = puffin.to_polars_api(df, api_version="0.20") result = dfx.with_columns( c=plx.col("a") + plx.col("b"), diff --git a/tpch/q1.py b/tpch/q1.py index 8cebfccda..fbd2dd9ab 100644 --- a/tpch/q1.py +++ b/tpch/q1.py @@ -1,7 +1,7 @@ # ruff: noqa from typing import Any from datetime import datetime -from polars_api_compat import to_polars_api, to_original_object +from puffin import to_polars_api, to_original_object import pandas as pd import polars import modin.pandas as mpd diff --git a/tpch/q2.py b/tpch/q2.py index 18a3b72ec..66dc965c3 100644 --- a/tpch/q2.py +++ b/tpch/q2.py @@ -4,8 +4,8 @@ import polars import pandas as pd -from polars_api_compat import to_original_object -from polars_api_compat import to_polars_api +from puffin import to_original_object +from puffin import to_polars_api polars.Config.set_tbl_cols(10) pd.set_option("display.max_columns", 10) diff --git a/tpch/q3.py b/tpch/q3.py index c1ec5c42d..b514441d7 100644 --- a/tpch/q3.py +++ b/tpch/q3.py @@ -6,8 +6,8 @@ import polars import pandas as pd -from polars_api_compat import to_original_object -from polars_api_compat import to_polars_api +from puffin import to_original_object +from puffin import to_polars_api import polars polars.Config.set_tbl_cols(10) diff --git a/tpch/q4.py b/tpch/q4.py index c66c57f1a..2d991060f 100644 --- a/tpch/q4.py +++ b/tpch/q4.py @@ -4,8 +4,8 @@ import polars -from polars_api_compat import to_original_object -from polars_api_compat import to_polars_api +from puffin import to_original_object +from puffin import to_polars_api Q_NUM = 4 diff --git a/tpch/q5.py b/tpch/q5.py index 5e3e516f0..9164510ab 100644 --- a/tpch/q5.py +++ b/tpch/q5.py @@ -4,8 +4,8 @@ import polars -from polars_api_compat import to_original_object -from polars_api_compat import to_polars_api +from puffin import to_original_object +from puffin import to_polars_api def q5(