Skip to content
This repository was archived by the owner on Oct 25, 2024. It is now read-only.

Commit c087c74

Browse files
authored
fix skip UT (#924)
1 parent 438e443 commit c087c74

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

intel_extension_for_transformers/neural_chat/tests/ci/api/test_chatbot_build_api.py

+3-3
Original file line numberDiff line numberDiff line change
@@ -80,7 +80,7 @@ def test_build_chatbot_with_audio_plugin(self):
8080
self.assertTrue(os.path.exists("./output_audio.wav"))
8181

8282
def test_build_chatbot_with_safety_checker_plugin(self):
83-
self.skipTest(self, "skip before fix the issue")
83+
self.skipTest("skip before fix the issue")
8484
plugins.safety_checker.enable = True
8585
pipeline_config = PipelineConfig(model_name_or_path="facebook/opt-125m",
8686
plugins=plugins)
@@ -91,7 +91,7 @@ def test_build_chatbot_with_safety_checker_plugin(self):
9191
self.assertTrue(response, "Your query contains sensitive words, please try another query.")
9292

9393
def test_build_chatbot_with_retrieval_plugin(self):
94-
self.skipTest(self, "skip before fix the issue")
94+
self.skipTest("skip before fix the issue")
9595
plugins.retrieval.enable = True
9696
plugins.retrieval.args["input_path"] = "../../../README.md"
9797
pipeline_config = PipelineConfig(model_name_or_path="facebook/opt-125m",
@@ -124,7 +124,7 @@ def test_build_chatbot_with_retrieval_plugin(self):
124124
self.assertIsNotNone(response)
125125

126126
def test_build_chatbot_with_retrieval_plugin_bge_int8(self):
127-
self.skipTest(self, "skip before fix the issue")
127+
self.skipTest("skip before fix the issue")
128128
if self.device != "cpu":
129129
self.skipTest("Only support Intel/bge-base-en-v1.5-sts-int8-static run on Intel CPU")
130130
plugins.retrieval.enable = True

intel_extension_for_transformers/neural_chat/tests/ci/api/test_inference.py

+4-4
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,7 @@ def tearDown(self) -> None:
3434
return super().tearDown()
3535

3636
def test_text_chat(self):
37-
self.skipTest(self, "skip before fix the issue")
37+
self.skipTest("skip before fix the issue")
3838
config = PipelineConfig(model_name_or_path="facebook/opt-125m")
3939
chatbot = build_chatbot(config)
4040
response = chatbot.predict("Tell me about Intel Xeon Scalable Processors.")
@@ -57,7 +57,7 @@ def test_retrieval(self):
5757
plugins.retrieval.enable = False
5858

5959
def test_retrieval_append(self):
60-
self.skipTest(self, "skip before fix the issue")
60+
self.skipTest("skip before fix the issue")
6161
plugins.retrieval.enable = True
6262
plugins.retrieval.args["append"] = True
6363
plugins.retrieval.args["input_path"] = "../assets/docs/"
@@ -81,7 +81,7 @@ def test_retrieval_append(self):
8181
plugins.retrieval.enable = False
8282

8383
def test_voice_chat(self):
84-
self.skipTest(self, "skip before fix the issue")
84+
self.skipTest("skip before fix the issue")
8585
plugins.tts.enable = True
8686
plugins.tts.args["output_audio_path"] = "./response.wav"
8787
pipeline_config = PipelineConfig(model_name_or_path="facebook/opt-125m",
@@ -103,7 +103,7 @@ def test_quantization(self):
103103
optimize_model(model=model, config=config)
104104

105105
def test_text_chat_stream(self):
106-
self.skipTest(self, "skip before fix the issue")
106+
self.skipTest("skip before fix the issue")
107107
config = PipelineConfig(model_name_or_path="facebook/opt-125m")
108108
chatbot = build_chatbot(config)
109109
stream_text = ""

0 commit comments

Comments
 (0)