Currently this explicitly calls an astype

return Series(counts, index=index, name="count").astype("Int64")

But with ArrowExtensionArray now able to support all pyarrow types, this should preserve the pyarrow type and return int64[pyarrow]

Comment From: lukemanley

Maybe somewhat related - do you think the ArrowExtensionArray comparison methods should preserve the pyarrow type and return boolean[pyarrow]?

Comment From: mroeschke

Good catch. Agreed _cmp_method should be changed to return "boolean[pyarrow]" instead of "Boolean" type. Since ArrowStringArray and ArrowExtensionArray are both marked as experimental it should be okay to change it for 2.0 if you're interested.

Comment From: lukemanley

Sounds good. I'll take a look