[BUG] test_test_example: ImportError: cannot import name 'MultiheadAttention' from 'zeta' (/home/v/.local/lib/python3.10/site-packages/zeta/__init__.py)

[BUG] test_test_example: ImportError: cannot import name 'MultiheadAttention' from 'zeta' (/home/v/.local/lib/python3.10/site-packages/zeta/__init__.py) #37