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