Skip to content

Commit

Permalink
fixup
Browse files Browse the repository at this point in the history
  • Loading branch information
MarcoGorelli committed Feb 19, 2024
1 parent cab4cdd commit fbcc22b
Show file tree
Hide file tree
Showing 8 changed files with 14 additions and 14 deletions.
4 changes: 2 additions & 2 deletions demo.py
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
2 changes: 1 addition & 1 deletion f.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")

Expand Down
4 changes: 2 additions & 2 deletions tests/test_common.py
Original file line number Diff line number Diff line change
@@ -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"),
Expand Down
2 changes: 1 addition & 1 deletion tpch/q1.py
Original file line number Diff line number Diff line change
@@ -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
Expand Down
4 changes: 2 additions & 2 deletions tpch/q2.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
4 changes: 2 additions & 2 deletions tpch/q3.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)
Expand Down
4 changes: 2 additions & 2 deletions tpch/q4.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
4 changes: 2 additions & 2 deletions tpch/q5.py
Original file line number Diff line number Diff line change
Expand Up @@ -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(
Expand Down

0 comments on commit fbcc22b

Please sign in to comment.