Skip to content

Remove jaxtyping dependency#121

Merged
Balandat merged 4 commits intocornellius-gp:mainfrom
Balandat:remove_jaxtyping
Feb 3, 2026
Merged

Remove jaxtyping dependency#121
Balandat merged 4 commits intocornellius-gp:mainfrom
Balandat:remove_jaxtyping

Conversation

@Balandat
Copy link
Collaborator

@Balandat Balandat commented Feb 3, 2026

Gets rid of the jaxtyping dependency as proposed in #113 (due to limited value and causing dependency issues in some places).

Summary:
Gets rid of the jaxtyping dependency as proposed in cornellius-gp#113 (due to limited value and causing dependency issues in some places).

Test Plan:
unit tests
@Balandat Balandat requested a review from gpleiss February 3, 2026 01:03
@Balandat
Copy link
Collaborator Author

Balandat commented Feb 3, 2026

run_small_type_checked_test_suite shows up as expected but of course won't complete since removed

@Balandat Balandat requested a review from kayween February 3, 2026 01:17
Balandat added a commit to Balandat/gpytorch that referenced this pull request Feb 3, 2026
cornellius-gp/linear_operator#121 removes jaxtyping from `linear_operator`, this removes the only occurrance also from gpytorch.
@jacobrgardner jacobrgardner self-requested a review February 3, 2026 14:57
Copy link
Member

@jacobrgardner jacobrgardner left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

This was a pretty big one of very "samey" changes but I scrolled through slowly and they all looked right to me.

Comment on lines +1 to +3
# Claude
CLAUDE.md

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I am assuming this is the prompt for Claude? Let's remove it.

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I guess I'm adding it here so the prompt doesn't get included when people use claude, do you want me to remove it from .gitignore so we can consider committing a version of the prompt?

Copy link
Collaborator Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Let me merge this for now; we can remove CLAUDE.md from .gitignore in a separate PR if we think that's the right move.

Copy link
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I don't think we should commit the prompt.

I initially thought these lines were added by mistake (since they seemed to be unrelated to this PR), and so I wanted to flagged these lines. But it looks like this gitignore entry was added intentionally. So we are all set here!

@kayween
Copy link
Collaborator

kayween commented Feb 3, 2026

Looks good to me as well.

@Balandat Balandat merged commit cd6ec0d into cornellius-gp:main Feb 3, 2026
4 checks passed
@Balandat Balandat deleted the remove_jaxtyping branch February 3, 2026 17:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants