fix [BUG] test_test_example: ImportError: cannot import name 'Multihe…

…adAttention' from 'zeta' (/home/v/.local/lib/python3.10/site-packages/zeta/__init__.py) #44
kyegomez · Dec 23, 2023 · 12d3c88 · 12d3c88
1 parent 3938501
commit 12d3c88
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_test_example.py b/tests/test_test_example.py
@@ -2,7 +2,7 @@
 import unittest
 import torch
 
-from zeta import MultiheadAttention
+from zeta.nn.attention import MultiheadAttention
 
 
 class TestMultiheadAttention(unittest.TestCase):