fix import path

ModelCloud · Nov 24, 2024 · 2d2d2fc · 2d2d2fc
1 parent 44f2d1a
commit 2d2d2fc
Show file tree

Hide file tree

Showing 32 changed files with 32 additions and 32 deletions.
diff --git a/tests/models/test_baichuan.py b/tests/models/test_baichuan.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestBaiChuan(ModelTest):

diff --git a/tests/models/test_bloom.py b/tests/models/test_bloom.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestBloom(ModelTest):

diff --git a/tests/models/test_chatglm.py b/tests/models/test_chatglm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 # The official THUDM/chatglm3-6b's tokenization_chatglm.py has compatibility issues with transformers.

diff --git a/tests/models/test_codegen.py b/tests/models/test_codegen.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestCodeGen(ModelTest):

diff --git a/tests/models/test_cohere.py b/tests/models/test_cohere.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestCohere(ModelTest):

diff --git a/tests/models/test_deci.py b/tests/models/test_deci.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestDeci(ModelTest):

diff --git a/tests/models/test_deepseekv2_lite.py b/tests/models/test_deepseekv2_lite.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestDeepseekV2Lite(ModelTest):

diff --git a/tests/models/test_exaone.py b/tests/models/test_exaone.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestExaone(ModelTest):

diff --git a/tests/models/test_falcon.py b/tests/models/test_falcon.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestFalcon(ModelTest):

diff --git a/tests/models/test_gemma.py b/tests/models/test_gemma.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestGemma(ModelTest):

diff --git a/tests/models/test_glm.py b/tests/models/test_glm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestGlm(ModelTest):

diff --git a/tests/models/test_gpt2.py b/tests/models/test_gpt2.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestGpt2(ModelTest):

diff --git a/tests/models/test_gptbigcode.py b/tests/models/test_gptbigcode.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestGptBigCode(ModelTest):

diff --git a/tests/models/test_gptj.py b/tests/models/test_gptj.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestGptJ(ModelTest):

diff --git a/tests/models/test_gptneox.py b/tests/models/test_gptneox.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 # TODO, this model requires 24G vram at least

diff --git a/tests/models/test_granite.py b/tests/models/test_granite.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestGranite(ModelTest):

diff --git a/tests/models/test_internlm.py b/tests/models/test_internlm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestInternlm(ModelTest):

diff --git a/tests/models/test_internlm2_5.py b/tests/models/test_internlm2_5.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestInternlm2_5(ModelTest):

diff --git a/tests/models/test_llama3_2.py b/tests/models/test_llama3_2.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestLlama3_2(ModelTest):

diff --git a/tests/models/test_longllama.py b/tests/models/test_longllama.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestLongLlama(ModelTest):

diff --git a/tests/models/test_minicpm.py b/tests/models/test_minicpm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestMiniCpm(ModelTest):

diff --git a/tests/models/test_mistral.py b/tests/models/test_mistral.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestMistral(ModelTest):

diff --git a/tests/models/test_mixtral.py b/tests/models/test_mixtral.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestMixtral(ModelTest):

diff --git a/tests/models/test_mpt.py b/tests/models/test_mpt.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestMpt(ModelTest):

diff --git a/tests/models/test_opt.py b/tests/models/test_opt.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestOpt(ModelTest):

diff --git a/tests/models/test_phi_3.py b/tests/models/test_phi_3.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestPhi_3(ModelTest):

diff --git a/tests/models/test_qwen2_5.py b/tests/models/test_qwen2_5.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestQwen2_5(ModelTest):

diff --git a/tests/models/test_stablelm.py b/tests/models/test_stablelm.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestStablelm(ModelTest):

diff --git a/tests/models/test_starcode2.py b/tests/models/test_starcode2.py
@@ -1,6 +1,6 @@
 import torch  # noqa: E402
 
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestStarCode2(ModelTest):

diff --git a/tests/models/test_tinyllama.py b/tests/models/test_tinyllama.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestTinyllama(ModelTest):

diff --git a/tests/models/test_xverse.py b/tests/models/test_xverse.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestXVerse(ModelTest):

diff --git a/tests/models/test_yi.py b/tests/models/test_yi.py
@@ -1,4 +1,4 @@
-from tests.model_test import ModelTest  # noqa: E402
+from ..model_test import ModelTest  # noqa: E402
 
 
 class TestYi(ModelTest):