Skip to content

[FEA]: Add type annotations and export for downstream packages #2084

@TomAugspurger

Description

@TomAugspurger

Is your feature request related to a problem? Please describe.

I had a bug in my usage of RMM that would have been caught by type annotations (rapidsai/cudf#20154 (comment)).

Describe the solution you'd like

Type annotations for all the public APIs in RMM. This would mean inline annotations in the python files and .pyi files for Cython (IIUC, that's still the best way to do things).

I believe that the packaging needs to be updated to include a py.typed so that downstream projects can use them.

Describe alternatives you've considered
N/A

Additional context
N/A

Metadata

Metadata

Assignees

No one assigned

    Labels

    Type

    No type

    Projects

    Status

    To-do

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions