Files
ray-project--ray/python/ray/air/util/tensor_extensions/arrow.py
T
2026-07-13 13:17:40 +08:00

10 lines
408 B
Python

# NOTE: We provide these as aliases to maintain compatibility with older version
# of Arrow `PyExtensionType` that relies on picked class references that
# reference `ray.air.util.tensor_extensions.arrow.*` classes
from ray.data._internal.tensor_extensions.arrow import (
ArrowTensorType, # noqa: F401
ArrowTensorTypeV2, # noqa: F401
ArrowVariableShapedTensorType, # noqa: F401
)