Ray Data implements PyArrow and pandas data type extensions to natively support Python objects and tensors.
Even though these extensions are only used by Ray Data, they live in the ray.air subpackage (this is due to historical reasons).
To make the package structure easier to understand, we should move these files to the ray.data subpackage and update imports accordingly:
- ray.air.util.objection_extensions -> ray.data._internal.object_extensions
- ray.air.util.tensor_extensions -> ray.data._internal.tensor_extensions
Ray Data implements PyArrow and pandas data type extensions to natively support Python objects and tensors.
Even though these extensions are only used by Ray Data, they live in the ray.air subpackage (this is due to historical reasons).
To make the package structure easier to understand, we should move these files to the ray.data subpackage and update imports accordingly: