I believe there is WIP to get python type annotations for arrays/tensors shape, but it's not a thing yet, indeed.