From f0aaca461b38eea2830b5025504ba55f64252f3e Mon Sep 17 00:00:00 2001 From: suluyan Date: Tue, 21 Jan 2025 12:36:57 +0800 Subject: [PATCH] fix test case --- tests/tools/test_to_ollama.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/tools/test_to_ollama.py b/tests/tools/test_to_ollama.py index 0c64729c..67c1c5d4 100644 --- a/tests/tools/test_to_ollama.py +++ b/tests/tools/test_to_ollama.py @@ -16,7 +16,7 @@ def _test_check_tmpl_type(model, tmpl_type, gguf_meta={}): class TestToOllama(unittest.TestCase): - @unittest.skip#Unless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') def test_load_template(self): template = TemplateLoader.load_by_model_id( 'LLM-Research/Meta-Llama-3-8B-Instruct') @@ -86,7 +86,7 @@ class TestToOllama(unittest.TestCase): 'LLM-Research/Phi-3-128k-instruct-GGUF') self.assertTrue(template.template_type == TemplateType.phi3) - @unittest.skip#Unless(test_level() >= 0, 'skip test in current test level') + @unittest.skipUnless(test_level() >= 0, 'skip test in current test level') def test_load_ollama(self): ollama = TemplateLoader.to_ollama( 'LLM-Research/Meta-Llama-3.1-8B-Instruct-GGUF')