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]