Show typing.Annotated info in tooltips

[lostmsu]

I work with PyTorch and want to be able to annotate tensor types with shapes.

I tried jaxtyping module like this:

def foo(x: jaxtyping.Float[torch.Tensor, "B T E") -> jaxtyping.Float[torch.Tensor, "B T 1"]: ...

Unfortunately, in VS Code when I hover over foo callsite or the definition, VS Code loses extra annotations:

def foo(x: Tensor) -> Tensor: ...

[github-actions[bot]]

Hi @lostmsu, I'm an AI Support assistant here to help with your issue. While the team reviews your request, I wanted to provide some possible tips and documentation that might help you in the meantime.

---

Similar Issues (FYI only, not solutions)

The following issues may or may not be related to your report. They are provided for reference only, as they may describe similar symptoms, affect the same feature, or address related concerns, but may not be the same issue.

• Pylance only displays the base type for typing.Annotated in tooltips and hides metadata such as shape annotations. Display metadata from typing.Annotated for function parameters #2306
• Hover shows Any for jaxtyping Annotated types instead of the full metadata (e.g., tensor shapes). Annotated results in Any type annotation when used with VSCODE pyright#6525

---

The team will respond to your issue shortly. Please note this is a trial feature and may not be fully accurate. I hope these suggestions are helpful in the meantime. If this comment helped you, please give it a 👍. If the suggestion was not helpful or was incorrect, please give it a 👎. Your feedback helps us improve!

[rchiodo]

Thanks for the suggestion. Transferring this to a discussion item for up votes.