Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Include jaxtyping to allow for Tensor/LinearOperator typehints with sizes. #2543

Merged
merged 1 commit into from
Jul 2, 2024
Merged

Include jaxtyping to allow for Tensor/LinearOperator typehints with sizes. #2543

merged 1 commit into from
Jul 2, 2024

Conversation

gpleiss
Copy link
Member

@gpleiss gpleiss commented Jul 2, 2024

Using the same trick in LinearOperator, sized Tensor/LinearOperator typehints are automatically included in the documentation.

@gpleiss
Include jaxtyping to allow for Tensor/LinearOperator typehints with s…
eea6549 
…izes.

Using the same trick in LinearOperator, sized Tensor/LinearOperator
typehints are automatically included in the documentation.
@gpleiss
Copy link
Member Author

gpleiss commented Jul 2, 2024

It's worth noting that jaxtyping is already implicitly required by gpytorch, since it is a requirement of linear_operator. This PR simply makes it possible to get pretty formatting of jaxtype'd TypeHints in the gpytorch documentation, following what we have in the linear_opeartor documentation.

@gpleiss gpleiss merged commit 9ff4640 into develop Jul 2, 2024
7 checks passed
@gpleiss gpleiss deleted the jaxtyping branch July 2, 2024 23:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@gpleiss