Skip to content

Commit 3dd4fa9

Browse files
fix: Import for arrow_orc_datasource
1 parent fd051b1 commit 3dd4fa9

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

awswrangler/distributed/ray/datasources/arrow_orc_datasource.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
from typing import Any, Dict, List, Optional
33

44
import pyarrow as pa
5-
from pyarrow import orc
65
from ray.data.block import BlockAccessor
76

87
from awswrangler._arrow import _add_table_partitions, _df_to_table
@@ -21,6 +20,8 @@ def _read_file( # type: ignore[override]
2120
path_root: str,
2221
**reader_args: Any,
2322
) -> pa.Table:
23+
from pyarrow import orc
24+
2425
columns: Optional[List[str]] = reader_args.get("columns", None)
2526

2627
table: pa.Table = orc.read_table(f, columns=columns)
@@ -48,6 +49,8 @@ def _write_block( # type: ignore[override]
4849
pandas_kwargs: Optional[Dict[str, Any]],
4950
**writer_args: Any,
5051
) -> None:
52+
from pyarrow import orc
53+
5154
schema: Optional[pa.schema] = writer_args.get("schema", None)
5255
dtype: Optional[Dict[str, str]] = writer_args.get("dtype", None)
5356
index: bool = writer_args.get("index", False)

0 commit comments

Comments
 (0)