Skip to content

Commit 09a0bfa

Browse files
Fix pyarrow.parquet import in tests (#952)
* Resolves #951
1 parent 88b1bda commit 09a0bfa

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

modin/pandas/test/test_io.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -5,6 +5,7 @@
55
from modin.pandas.utils import to_pandas
66
from pathlib import Path
77
import pyarrow as pa
8+
import pyarrow.parquet as pq
89
import os
910
import shutil
1011
import sqlalchemy as sa
@@ -80,7 +81,7 @@ def _make_parquet_file(
8081
else:
8182
os.mkdir(TEST_PARQUET_FILENAME)
8283
table = pa.Table.from_pandas(df)
83-
pa.parquet.write_to_dataset(table, root_path=TEST_PARQUET_FILENAME)
84+
pq.write_to_dataset(table, root_path=TEST_PARQUET_FILENAME)
8485
elif len(partitioned_columns) > 0:
8586
df.to_parquet(TEST_PARQUET_FILENAME, partition_cols=partitioned_columns)
8687
else:

0 commit comments

Comments
 (0)