From 1d657f7aaab6a0ac6f594d806a6731e04a402594 Mon Sep 17 00:00:00 2001
From: vyomakesh09 <vyomkeshkesh@gmail.com>
Date: Sat, 23 Dec 2023 01:09:26 +0000
Subject: [PATCH] fix [BUG] test_test_example: ImportError: cannot import name
 'MultiheadAttention' from 'zeta'
 (/home/v/.local/lib/python3.10/site-packages/zeta/__init__.py) #44

---
 tests/test_test_example.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_test_example.py b/tests/test_test_example.py
index fbcfa709..0e6ad8e2 100644
--- 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):