From fb1b25507cbebc25f6563927d0e74a11d017fcb5 Mon Sep 17 00:00:00 2001
From: Falko Habel <falko.habel@gmx.de>
Date: Fri, 28 Mar 2025 16:10:22 +0100
Subject: [PATCH] added Expert to model to fix test

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

diff --git a/tests/model/test_aiia.py b/tests/model/test_aiia.py
index 340537b..bffa616 100644
--- a/tests/model/test_aiia.py
+++ b/tests/model/test_aiia.py
@@ -1,6 +1,6 @@
 import os
 import torch
-from aiia import AIIABase, AIIABaseShared, AIIAExpert, AIIAmoe, AIIAchunked, AIIAConfig, AIIASparseMoe
+from aiia.model.Model import AIIABase, AIIABaseShared, AIIAExpert, AIIAmoe, AIIAchunked, AIIAConfig, AIIASparseMoe
 
 def test_aiiabase_creation():
     config = AIIAConfig()