diff --git a/pyproject.toml b/pyproject.toml index ae15a8a4b..4132f0a25 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -73,6 +73,7 @@ dev = [ "duckdb>=1.0", "ipython[kernel]", "pandas>=1.1.3", + "pyarrow-stubs", "pytest", "pytest-cov", "pytest-xdist[psutil]~=3.5", diff --git a/tests/utils/test_utils.py b/tests/utils/test_utils.py index c3b329cf0..2e8ae1214 100644 --- a/tests/utils/test_utils.py +++ b/tests/utils/test_utils.py @@ -140,7 +140,7 @@ def test_sanitize_pyarrow_table_columns() -> None: pa_table = pa.Table.from_pandas( df, pa.schema( - [ + ( pa.field("s", pa.string()), pa.field("f", pa.float64()), pa.field("i", pa.int64()), @@ -148,7 +148,7 @@ def test_sanitize_pyarrow_table_columns() -> None: pa.field("d", pa.date32()), pa.field("c", pa.dictionary(pa.int8(), pa.string())), pa.field("p", pa.timestamp("ns", tz="UTC")), - ] + ) ), ) sanitized = sanitize_narwhals_dataframe(nw.from_native(pa_table, eager_only=True))