Skip to content

Conversation

@zigzagcai
Copy link
Contributor

@zigzagcai zigzagcai commented Jul 16, 2024

Dear author,

We noticed that the unittest test_selective_state_update_with_heads was added in some recent commit, but when we tried to run pytest tests/, we found an error occured:

NameError: name 'repeat' is not defined.

So we add the missing import part of from einops import repeat, and it works.
Hope it helps!

@zigzagcai zigzagcai changed the title Fix test for test_selective_state_update_with_heads Fix unittest for test_selective_state_update_with_heads Jul 16, 2024
@tridao tridao merged commit c0a00bd into state-spaces:main Jul 16, 2024
@zigzagcai zigzagcai deleted the fix-tests branch July 20, 2024 12:48
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.

2 participants