fixup

Author: MarcoGorelli

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"

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")
 

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"),

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

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)

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)

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
 

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(