refactor(pyarrow-format): avoid constructing unnecessary array to pro…

…duce a scalar
ibis-project · Sep 24, 2024 · 92e6657 · 92e6657
1 parent c8c37dd
commit 92e6657
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/ibis/formats/pyarrow.py b/ibis/formats/pyarrow.py
@@ -304,9 +304,7 @@ def convert_scalar(cls, scalar: pa.Scalar, dtype: dt.DataType) -> pa.Scalar:
             except pa.ArrowNotImplementedError:
                 # pyarrow doesn't support some scalar casts that are supported
                 # when using arrays or tables
-                return pa.array([scalar.as_py()], type=scalar_type).cast(desired_type)[
-                    0
-                ]
+                return pa.scalar(scalar.as_py(), type=scalar_type).cast(desired_type)
         else:
             return scalar