Skip to content

fix: clarify assert_shape error when bare Ellipsis is passed#439

Open
PewterZz wants to merge 1 commit intogoogle-deepmind:mainfrom
PewterZz:fix/assert-shape-ellipsis-error
Open

fix: clarify assert_shape error when bare Ellipsis is passed#439
PewterZz wants to merge 1 commit intogoogle-deepmind:mainfrom
PewterZz:fix/assert-shape-ellipsis-error

Conversation

@PewterZz
Copy link
Copy Markdown

Fixes #437

When passing bare ... to assert_shape, the error message says "got Ellipsis" which doesn't help the user understand what went wrong. Added a specific check that tells them ... needs to be wrapped as (...,).

Tested locally.

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.

Improve assert_shape error message when Ellipsis is used as shape matcher

1 participant