geshi

Author: huhu-dsy

pandas/core/arrays/arrow/accessors.py

@@ -168,9 +168,11 @@ def __getitem__(self, key: int | slice) -> Series:
                         if not arr.is_null()[i].as_py():
                             current_length = lengths[i].as_py()
                             if current_length == 0:
-                                raise IndexError(f"Index {key} is out of bounds: should be in [0, 0)")
+                                raise IndexError(
+f"Index {key} is out of bounds: should be in [0, 0)")
                             if current_length < abs(key):
-                                raise IndexError(f"Index {key} is out of bounds: should be in [{-current_length}, {current_length})")
+                                raise IndexError(
+f"Index {key} is out of bounds: should be in [{-current_length}, {current_length})")
                 chunks = arr.chunks if isinstance(arr, pa.ChunkedArray) else [arr]
                 all_results = []
                 for chunk in chunks:
@@ -188,7 +190,9 @@ def __getitem__(self, key: int | slice) -> Series:
                     else:
                         chunk_result = taken_values
                     all_results.append(chunk_result)
-                result_values = pa.concat_arrays(all_results) if all_results else pa.array([],type=arr.type.value_type)
+                result_values = pa.concat_arrays(
+                    all_results) if all_results else pa.array([],
+                                                              type=arr.type.value_type)
                 return Series(
                     result_values,
                     dtype=ArrowDtype(result_values.type),

pandas/tests/series/accessors/test_list_accessor.py

@@ -41,7 +41,8 @@ def test_list_getitem_negative_index():
     tm.assert_series_equal(actual, expected)
 
     ser_empty = Series([[]], dtype=ArrowDtype(pa.list_(pa.int64())))
-    with pytest.raises(IndexError, match="Index -1 is out of bounds: should be in \\[0, 0\\)"):
+    with pytest.raises(
+            IndexError, match="Index -1 is out of bounds: should be in \\[0, 0\\)"):
         ser_empty.list[-1]
 
     ser_mixed = Series([[1], [1, 2], [1, 2, 3]], dtype=ArrowDtype(pa.list_(pa.int64())))