cuddly-jelly-27016
05/29/2025, 12:10 AMnp.ndarray type, but it doesn't support the np.typing.NDArray generic: https://numpy.org/devdocs/reference/typing.html#numpy.typing.NDArray
This would enable flytekit users to express the actual datatypes of an ndarray.
### Goal: What should the final outcome look like, ideally?
The user-facing API would be simple:
import numpy
import numpy.typing as npt
@task
def t1(input: npt.NDArray[np.float64]) -> npt.NDArray[np.int64]:
...
### Describe alternatives you've considered
Keeping the status quo, only supporting np.ndarray
### Propose: Link/Inline OR Additional context
No response
### Are you sure this issue hasn't been raised already?
• Yes
### Have you read the Code of Conduct?
• Yes
flyteorg/flytecuddly-jelly-27016
05/29/2025, 12:10 AM