Update python/tests/api/logger/test_logger_polars.py

whylabs · Nov 7, 2024 · 09d1ad1 · 09d1ad1
1 parent e45edbe
commit 09d1ad1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/tests/api/logger/test_logger_polars.py b/python/tests/api/logger/test_logger_polars.py
@@ -56,8 +56,9 @@ def test_basic_log() -> None:
     assert profile._columns["col3"]._schema.dtype == pl.Utf8
 
 
-def test_lending_club(lending_club_df: pl.DataFrame) -> None:
-    res = why.log(lending_club_df)
+def test_lending_club(lending_club_df: pd.DataFrame) -> None:
+    df = pl.from_pandas(lending_club_df)
+    res = why.log(df)
     view = res.view()
     df = view.to_pandas()
     assert len(df) == 151