added Expert to model to fix test

2025-03-28 16:10:22 +01:00 · 2025-03-28 16:10:22 +01:00 · fb1b25507c
parent 5987a130f6
commit fb1b25507c
1 changed files with 1 additions and 1 deletions
--- 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()