Skip to content
New issue

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

#

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

Already on GitHub? # 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
# for free to join this conversation on GitHub. Already have an account? # 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