We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 88b1bda commit 09a0bfaCopy full SHA for 09a0bfa
modin/pandas/test/test_io.py
@@ -5,6 +5,7 @@
5
from modin.pandas.utils import to_pandas
6
from pathlib import Path
7
import pyarrow as pa
8
+import pyarrow.parquet as pq
9
import os
10
import shutil
11
import sqlalchemy as sa
@@ -80,7 +81,7 @@ def _make_parquet_file(
80
81
else:
82
os.mkdir(TEST_PARQUET_FILENAME)
83
table = pa.Table.from_pandas(df)
- pa.parquet.write_to_dataset(table, root_path=TEST_PARQUET_FILENAME)
84
+ pq.write_to_dataset(table, root_path=TEST_PARQUET_FILENAME)
85
elif len(partitioned_columns) > 0:
86
df.to_parquet(TEST_PARQUET_FILENAME, partition_cols=partitioned_columns)
87
0 commit comments