Yeah, overall I actually think TransformerLens does pretty well on type hints (and even better with v3 from what I’ve seen so far). run_with_hooks was just a particularly good example of what I’m talking about considering it’s one of the main entry points for the library, and the fix would be super easy.
Yeah, overall I actually think TransformerLens does pretty well on type hints (and even better with v3 from what I’ve seen so far).
run_with_hookswas just a particularly good example of what I’m talking about considering it’s one of the main entry points for the library, and the fix would be super easy.