**Describe the bug** Testing in geodatafusion, ```py from datafusion import SessionContext, udf, udaf from geodatafusion import native ctx.register_udaf(udaf(native.Extent())) ``` This works at runtime but does not type check. > No overloads for "udaf" match the provided arguments **To Reproduce** Steps to reproduce the behavior: **Expected behavior** Should type check. **Additional context** Add any other context about the problem here.