Skip to content

fix mypy errors in einops.layers.tensorflow and add missing annotations - #423

Open
jorenham wants to merge 2 commits into
arogozhnikov:mainfrom
jorenham:typing/layers.tensorflow
Open

fix mypy errors in einops.layers.tensorflow and add missing annotations#423
jorenham wants to merge 2 commits into
arogozhnikov:mainfrom
jorenham:typing/layers.tensorflow

Conversation

@jorenham

@jorenham jorenham commented May 5, 2026

Copy link
Copy Markdown
Contributor

Tensorflow has pretty solid static typing support nowadays, so that makes it possible to make the einops tensorflow layers nice and generic.

This also resolves all mypy errors in einops.layers.tensorflow, so say goodbye to those squiggly lines.

And as per usual, I've also filled in the annotations of the public typables, which in this case increases the type coverage of einops by a juicy 9.03%.

@jorenham

jorenham commented May 5, 2026

Copy link
Copy Markdown
Contributor Author

Oh that's annoying. Apparently tensorflow's Layer is only generic in the stubs, not at runtime...

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.

1 participant