wehub/ray-project--ray
1
0
Fork 0
Code Issues Pull Requests Actions Deployments Models Agent Notes Packages Projects Releases Wiki Activity

chore: import upstream snapshot with attribution

Browse Source
This commit is contained in:
wehub-resource-sync
2026-07-13 13:17:40 +08:00
commit f1825c8ceb
10096 changed files with 2364182 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
python/ray/data/extensions/__init__.py
+49
View File
@@ -0,0 +1,49 @@
from ray.data._internal.tensor_extensions.arrow import (
ArrowTensorTypeV2,
FixedShapeTensorFormat,
FixedShapeTensorType,
create_arrow_fixed_shape_tensor_type,
get_arrow_extension_tensor_types,
)
from ray.data.extensions.object_extension import (
ArrowPythonObjectArray,
ArrowPythonObjectScalar,
ArrowPythonObjectType,
PythonObjectArray,
PythonObjectDtype,
)
from ray.data.extensions.tensor_extension import (
ArrowConversionError,
ArrowTensorArray,
ArrowTensorType,
ArrowVariableShapedTensorArray,
ArrowVariableShapedTensorType,
TensorArray,
TensorArrayElement,
TensorDtype,
column_needs_tensor_extension,
)
__all__ = [
# Tensor array extension.
"TensorDtype",
"TensorArray",
"TensorArrayElement",
"ArrowTensorType",
"ArrowTensorTypeV2",
"ArrowTensorArray",
"ArrowVariableShapedTensorType",
"ArrowVariableShapedTensorArray",
"column_needs_tensor_extension",
"ArrowConversionError",
"FixedShapeTensorType",
"FixedShapeTensorFormat",
"create_arrow_fixed_shape_tensor_type",
# Object array extension
"ArrowPythonObjectArray",
"ArrowPythonObjectType",
"ArrowPythonObjectScalar",
"PythonObjectArray",
"PythonObjectDtype",
"get_arrow_extension_tensor_types",
]