patrick-kidger / patrick-kidger/jaxtyping
Failed to compile with Union
Open
Nobody has claimed this yet.
question
- Dominant language
- Python
- Stars
- 1.9k
- Forks
- 96
- PR merge metrics
- No merged PRs in 30d
Description
I have a function with signature:
from typing import Union
from jaxtyping import Float, jaxtyped
from typeguard import typechecked
@jaxtyped
@typechecked
def fn(x: Union[Float[torch.Tensor, "B F"], int]): ...
which doesn't work. It fails to compile the AST
Contributor guide
First steps
- Read the whole issue, then the project's contributing guide.
- Comment on the issue to say you are picking it up — it saves two people doing the same work.
- Fork the repository and make your change on a branch.
- Open a pull request that references the issue number.
Research direction
Start by reproducing the AST compilation failure with the Union annotation and decorator stack shown in the issue. Trace the annotation handling until the failure is located; done means this function compiles and its Union annotation is handled without the reported error.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- python
- Domain
- tooling
- Issue type
- Bug
- Difficulty
- 4/5
- Estimated time
- 3-5 days
- Activity status
- Stale
- Clarity
- Needs clarification
- Newbie friendliness
- 25/100