From 2e1e184a3b9b084d3cc297733296dd9116d00b0e Mon Sep 17 00:00:00 2001 From: CSY Date: Sun, 24 Nov 2024 19:05:37 +0800 Subject: [PATCH] remove .. --- tests/models/test_baichuan.py | 2 +- tests/models/test_bloom.py | 2 +- tests/models/test_chatglm.py | 2 +- tests/models/test_codegen.py | 2 +- tests/models/test_cohere.py | 2 +- tests/models/test_deci.py | 2 +- tests/models/test_exaone.py | 2 +- tests/models/test_falcon.py | 2 +- tests/models/test_gemma.py | 2 +- tests/models/test_glm.py | 2 +- tests/models/test_gpt2.py | 2 +- tests/models/test_gptbigcode.py | 2 +- tests/models/test_gptj.py | 2 +- tests/models/test_gptneox.py | 2 +- tests/models/test_granite.py | 2 +- tests/models/test_internlm.py | 2 +- tests/models/test_internlm2_5.py | 2 +- tests/models/test_llama3_2.py | 2 +- tests/models/test_longllama.py | 2 +- tests/models/test_minicpm.py | 2 +- tests/models/test_mistral.py | 2 +- tests/models/test_mixtral.py | 2 +- tests/models/test_mpt.py | 2 +- tests/models/test_opt.py | 2 +- tests/models/test_phi_3.py | 2 +- tests/models/test_qwen2_5.py | 2 +- tests/models/test_stablelm.py | 2 +- tests/models/test_starcode2.py | 2 +- tests/models/test_tinyllama.py | 2 +- tests/models/test_xverse.py | 2 +- tests/models/test_yi.py | 2 +- 31 files changed, 31 insertions(+), 31 deletions(-) diff --git a/tests/models/test_baichuan.py b/tests/models/test_baichuan.py index 1481536e5..1a533b3a4 100644 --- a/tests/models/test_baichuan.py +++ b/tests/models/test_baichuan.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestBaiChuan(ModelTest): diff --git a/tests/models/test_bloom.py b/tests/models/test_bloom.py index 2cfa94369..b3113d8a5 100644 --- a/tests/models/test_bloom.py +++ b/tests/models/test_bloom.py @@ -6,7 +6,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestBloom(ModelTest): diff --git a/tests/models/test_chatglm.py b/tests/models/test_chatglm.py index 8cdf1347e..54dabc633 100644 --- a/tests/models/test_chatglm.py +++ b/tests/models/test_chatglm.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest # 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 index af8a780b3..05c0cb1a1 100644 --- a/tests/models/test_codegen.py +++ b/tests/models/test_codegen.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestCodeGen(ModelTest): diff --git a/tests/models/test_cohere.py b/tests/models/test_cohere.py index 4f6951972..9e5c2e2f1 100644 --- a/tests/models/test_cohere.py +++ b/tests/models/test_cohere.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestCohere(ModelTest): diff --git a/tests/models/test_deci.py b/tests/models/test_deci.py index c68989e25..95d153e66 100644 --- a/tests/models/test_deci.py +++ b/tests/models/test_deci.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestDeci(ModelTest): diff --git a/tests/models/test_exaone.py b/tests/models/test_exaone.py index b9994ed70..f794f8f1f 100644 --- a/tests/models/test_exaone.py +++ b/tests/models/test_exaone.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestExaone(ModelTest): diff --git a/tests/models/test_falcon.py b/tests/models/test_falcon.py index ee1eef579..193851e4b 100644 --- a/tests/models/test_falcon.py +++ b/tests/models/test_falcon.py @@ -6,7 +6,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestFalcon(ModelTest): diff --git a/tests/models/test_gemma.py b/tests/models/test_gemma.py index 95a56322e..25a3639dc 100644 --- a/tests/models/test_gemma.py +++ b/tests/models/test_gemma.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestGemma(ModelTest): diff --git a/tests/models/test_glm.py b/tests/models/test_glm.py index a7f3690db..9ff2ed3f3 100644 --- a/tests/models/test_glm.py +++ b/tests/models/test_glm.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestGlm(ModelTest): diff --git a/tests/models/test_gpt2.py b/tests/models/test_gpt2.py index e18df996c..1aeb3c412 100644 --- a/tests/models/test_gpt2.py +++ b/tests/models/test_gpt2.py @@ -6,7 +6,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestGpt2(ModelTest): diff --git a/tests/models/test_gptbigcode.py b/tests/models/test_gptbigcode.py index ac1367422..1aaead4cb 100644 --- a/tests/models/test_gptbigcode.py +++ b/tests/models/test_gptbigcode.py @@ -6,7 +6,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestGptBigCode(ModelTest): diff --git a/tests/models/test_gptj.py b/tests/models/test_gptj.py index 78cbe880e..95326586f 100644 --- a/tests/models/test_gptj.py +++ b/tests/models/test_gptj.py @@ -6,7 +6,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestGptJ(ModelTest): diff --git a/tests/models/test_gptneox.py b/tests/models/test_gptneox.py index 2c2e4b7be..36c8626f5 100644 --- a/tests/models/test_gptneox.py +++ b/tests/models/test_gptneox.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest # TODO, this model requires 24G vram at least diff --git a/tests/models/test_granite.py b/tests/models/test_granite.py index 457acb1d8..ac94b7c51 100644 --- a/tests/models/test_granite.py +++ b/tests/models/test_granite.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestGranite(ModelTest): diff --git a/tests/models/test_internlm.py b/tests/models/test_internlm.py index c53f2ffa4..2fe93c13f 100644 --- a/tests/models/test_internlm.py +++ b/tests/models/test_internlm.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestInternlm(ModelTest): diff --git a/tests/models/test_internlm2_5.py b/tests/models/test_internlm2_5.py index 9fad10bad..ecfb8ea7e 100644 --- a/tests/models/test_internlm2_5.py +++ b/tests/models/test_internlm2_5.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestInternlm2_5(ModelTest): diff --git a/tests/models/test_llama3_2.py b/tests/models/test_llama3_2.py index 092e207ef..088dfdcf9 100644 --- a/tests/models/test_llama3_2.py +++ b/tests/models/test_llama3_2.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestLlama3_2(ModelTest): diff --git a/tests/models/test_longllama.py b/tests/models/test_longllama.py index fc0092cbc..29839db5f 100644 --- a/tests/models/test_longllama.py +++ b/tests/models/test_longllama.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestLongLlama(ModelTest): diff --git a/tests/models/test_minicpm.py b/tests/models/test_minicpm.py index 71904f555..9bb256651 100644 --- a/tests/models/test_minicpm.py +++ b/tests/models/test_minicpm.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestMiniCpm(ModelTest): diff --git a/tests/models/test_mistral.py b/tests/models/test_mistral.py index bf3e492b8..f9de5dd21 100644 --- a/tests/models/test_mistral.py +++ b/tests/models/test_mistral.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestMistral(ModelTest): diff --git a/tests/models/test_mixtral.py b/tests/models/test_mixtral.py index ef822099c..46f4e94fb 100644 --- a/tests/models/test_mixtral.py +++ b/tests/models/test_mixtral.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestMixtral(ModelTest): diff --git a/tests/models/test_mpt.py b/tests/models/test_mpt.py index 32d48525e..95a2ef041 100644 --- a/tests/models/test_mpt.py +++ b/tests/models/test_mpt.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestMpt(ModelTest): diff --git a/tests/models/test_opt.py b/tests/models/test_opt.py index b3126f2c4..5c7c1ebdc 100644 --- a/tests/models/test_opt.py +++ b/tests/models/test_opt.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestOpt(ModelTest): diff --git a/tests/models/test_phi_3.py b/tests/models/test_phi_3.py index 3409ae925..12f63bbe5 100644 --- a/tests/models/test_phi_3.py +++ b/tests/models/test_phi_3.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestPhi_3(ModelTest): diff --git a/tests/models/test_qwen2_5.py b/tests/models/test_qwen2_5.py index 56549fe18..043d16548 100644 --- a/tests/models/test_qwen2_5.py +++ b/tests/models/test_qwen2_5.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestQwen2_5(ModelTest): diff --git a/tests/models/test_stablelm.py b/tests/models/test_stablelm.py index 3ddd7b26e..d97c684b6 100644 --- a/tests/models/test_stablelm.py +++ b/tests/models/test_stablelm.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestStablelm(ModelTest): diff --git a/tests/models/test_starcode2.py b/tests/models/test_starcode2.py index b169e67a9..24398876f 100644 --- a/tests/models/test_starcode2.py +++ b/tests/models/test_starcode2.py @@ -6,7 +6,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestStarCode2(ModelTest): diff --git a/tests/models/test_tinyllama.py b/tests/models/test_tinyllama.py index 7122e78e7..beca51871 100644 --- a/tests/models/test_tinyllama.py +++ b/tests/models/test_tinyllama.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestTinyllama(ModelTest): diff --git a/tests/models/test_xverse.py b/tests/models/test_xverse.py index 2b4ea646f..62546bdf4 100644 --- a/tests/models/test_xverse.py +++ b/tests/models/test_xverse.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestXVerse(ModelTest): diff --git a/tests/models/test_yi.py b/tests/models/test_yi.py index 375bd8187..33b72cb35 100644 --- a/tests/models/test_yi.py +++ b/tests/models/test_yi.py @@ -4,7 +4,7 @@ parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), '..')) sys.path.insert(0, parent_dir) -from ..model_test import ModelTest # noqa: E402 +from model_test import ModelTest class TestYi(ModelTest):