Skip to content

Fix type hint to allow DataFusion PyCapsule provider into udaf function #1237

@kylebarron

Description

@kylebarron

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.

Metadata

Metadata

Assignees

No one assigned

    Labels

    bugSomething isn't working

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions