Skip to content

Conversation

@rostan-t
Copy link
Collaborator

@rostan-t rostan-t commented Nov 18, 2025

Category: New feature (non-breaking change which adds functionality)

Description:

Currently stub (*.pyi) files are only generated for "fn" and the legacy "ops" API. This PR adds stubs generation for the dynamic mode.

Additional information:

Affected modules and functionalities:

New __init__.pyi files get created during the build process.

Key points relevant for the review:

  • Does the IDE / language server complain when using the API normally?

  • Does the IDE / language server properly report incorrect types?

  • Does the IDE properly suggest code completion?

  • You can use typing.reveal_type to see what type the type checker infers. For example:

    example

Tests:

TODO

  • Existing tests apply
  • New tests added
    • Python tests
    • GTests
    • Benchmark
    • Other
  • N/A

Checklist

Documentation

DALI team only

Requirements

  • Implements new requirements
  • Affects existing requirements
  • N/A

REQ IDs: N/A

JIRA TASK: DALI-4439


@property
def __cuda_array_interface__(self) -> Any: ...
def __cuda_array_interface__(self) -> dict[str, Any]: ...

Check notice

Code scanning / CodeQL

Statement has no effect Note

This statement has no effect.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant