Describe the bug

Testing in geodatafusion,

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.