blob: 944f3390e31f3040eec72a1c7e5b1931325c5876 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
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())
|
