This commit is contained in:
2025-09-07 22:09:54 +02:00
parent e1b817252c
commit 2fc0d000b6
7796 changed files with 2159515 additions and 933 deletions

View File

@ -0,0 +1,18 @@
from __future__ import annotations
from typing import TYPE_CHECKING
import pyarrow as pa
from narwhals._arrow.utils import ArrowSeriesNamespace
if TYPE_CHECKING:
from narwhals._arrow.series import ArrowSeries
from narwhals._arrow.typing import Incomplete
class ArrowSeriesCatNamespace(ArrowSeriesNamespace):
def get_categories(self) -> ArrowSeries:
# NOTE: Should be `list[pa.DictionaryArray]`, but `DictionaryArray` has no attributes
chunks: Incomplete = self.native.chunks
return self.with_native(pa.concat_arrays(x.dictionary for x in chunks).unique())