From 651092d3a2516f690d3cd7bb1dadea121ef6a908 Mon Sep 17 00:00:00 2001 From: mae5357 Date: Mon, 13 May 2024 16:43:11 +0000 Subject: [PATCH] add metadata and allow_losing_tz --- src/pydantic_to_pyarrow/schema.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/pydantic_to_pyarrow/schema.py b/src/pydantic_to_pyarrow/schema.py index 12cfa8f..ba80ec3 100644 --- a/src/pydantic_to_pyarrow/schema.py +++ b/src/pydantic_to_pyarrow/schema.py @@ -116,8 +116,8 @@ def _get_dict_type( ) -> pa.DataType: key_type, value_type = get_args(field_type) return pa.map_( - _get_pyarrow_type(key_type, [], allow_losing_tz=False), - _get_pyarrow_type(value_type, [], allow_losing_tz=False), + _get_pyarrow_type(key_type, metadata, allow_losing_tz=allow_losing_tz), + _get_pyarrow_type(value_type, metadata, allow_losing_tz=allow_losing_tz), )