1212 StructuredDatasetEncoder
1313"""
1414
15+ import functools
16+
1517from flytekit .deck .renderer import ArrowRenderer , TopFrameRenderer
1618from flytekit .lazy_import .lazy_module import is_imported
1719from flytekit .loggers import logger
2527)
2628
2729
30+ @functools .lru_cache (maxsize = None )
2831def register_csv_handlers ():
2932 from .basic_dfs import CSVToPandasDecodingHandler , PandasToCSVEncodingHandler
3033
3134 StructuredDatasetTransformerEngine .register (PandasToCSVEncodingHandler (), default_format_for_type = True )
3235 StructuredDatasetTransformerEngine .register (CSVToPandasDecodingHandler (), default_format_for_type = True )
3336
3437
38+ @functools .lru_cache (maxsize = None )
3539def register_pandas_handlers ():
3640 import pandas as pd
3741
@@ -42,6 +46,7 @@ def register_pandas_handlers():
4246 StructuredDatasetTransformerEngine .register_renderer (pd .DataFrame , TopFrameRenderer ())
4347
4448
49+ @functools .lru_cache (maxsize = None )
4550def register_arrow_handlers ():
4651 import pyarrow as pa
4752
@@ -52,6 +57,7 @@ def register_arrow_handlers():
5257 StructuredDatasetTransformerEngine .register_renderer (pa .Table , ArrowRenderer ())
5358
5459
60+ @functools .lru_cache (maxsize = None )
5561def register_bigquery_handlers ():
5662 try :
5763 from .bigquery import (
@@ -72,6 +78,7 @@ def register_bigquery_handlers():
7278 )
7379
7480
81+ @functools .lru_cache (maxsize = None )
7582def register_snowflake_handlers ():
7683 try :
7784 from .snowflake import PandasToSnowflakeEncodingHandlers , SnowflakeToPandasDecodingHandler
0 commit comments