Skip to content

Commit f6825c2

Browse files
authored
fix: remove _default_metadata_providers (#3007)
1 parent 1b74dc0 commit f6825c2

File tree

1 file changed

+2
-4
lines changed

1 file changed

+2
-4
lines changed

awswrangler/distributed/ray/datasources/arrow_parquet_datasource.py

+2-4
Original file line numberDiff line numberDiff line change
@@ -30,11 +30,9 @@
3030
from ray.data.block import Block
3131
from ray.data.context import DataContext
3232
from ray.data.datasource import Datasource
33-
from ray.data.datasource._default_metadata_providers import (
34-
get_generic_metadata_provider,
35-
)
3633
from ray.data.datasource.datasource import ReadTask
3734
from ray.data.datasource.file_meta_provider import (
35+
DefaultFileMetadataProvider,
3836
_handle_read_os_error,
3937
)
4038
from ray.data.datasource.parquet_meta_provider import (
@@ -219,7 +217,7 @@ def __init__( # noqa: PLR0912,PLR0915
219217
# files. To avoid this, we expand the input paths with the default metadata
220218
# provider and then apply the partition filter or file extensions.
221219
if partition_filter is not None or file_extensions is not None:
222-
default_meta_provider = get_generic_metadata_provider(file_extensions=None)
220+
default_meta_provider = DefaultFileMetadataProvider()
223221
expanded_paths, _ = map(list, zip(*default_meta_provider.expand_paths(paths, filesystem)))
224222

225223
paths = list(expanded_paths)

0 commit comments

Comments
 (0)