lint

Signed-off-by: Alexey Kudinkin <ak@anyscale.com>
ray-project · Oct 1, 2024 · 65d7a07 · 65d7a07
1 parent 2de388c
commit 65d7a07
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/python/ray/air/util/transform_pyarrow.py b/python/ray/air/util/transform_pyarrow.py
@@ -20,7 +20,7 @@ def _concatenate_extension_column(ca: "pyarrow.ChunkedArray") -> "pyarrow.Array"
     """
     from ray.air.util.tensor_extensions.arrow import (
         ArrowTensorArray,
-        get_arrow_extension_tensor_types
+        get_arrow_extension_tensor_types,
     )
 
     if not _is_column_extension_type(ca):

diff --git a/python/ray/data/_internal/arrow_ops/transform_pyarrow.py b/python/ray/data/_internal/arrow_ops/transform_pyarrow.py
@@ -3,7 +3,7 @@
 from packaging.version import parse as parse_version
 
 from ray._private.utils import _get_pyarrow_version
-from ray.air.util.tensor_extensions.arrow import ArrowTensorTypeV2, get_arrow_extension_tensor_types
+from ray.air.util.tensor_extensions.arrow import ArrowTensorTypeV2
 
 try:
     import pyarrow

diff --git a/python/ray/data/extensions/__init__.py b/python/ray/data/extensions/__init__.py
@@ -41,5 +41,5 @@
     "PythonObjectArray",
     "PythonObjectDtype",
     "object_extension_type_allowed",
-    "get_arrow_extension_tensor_types"
+    "get_arrow_extension_tensor_types",
 ]