@@ -34,7 +34,7 @@ def tearDown(self) -> None:
34
34
return super ().tearDown ()
35
35
36
36
def test_text_chat (self ):
37
- self .skipTest (self , "skip before fix the issue" )
37
+ self .skipTest ("skip before fix the issue" )
38
38
config = PipelineConfig (model_name_or_path = "facebook/opt-125m" )
39
39
chatbot = build_chatbot (config )
40
40
response = chatbot .predict ("Tell me about Intel Xeon Scalable Processors." )
@@ -57,7 +57,7 @@ def test_retrieval(self):
57
57
plugins .retrieval .enable = False
58
58
59
59
def test_retrieval_append (self ):
60
- self .skipTest (self , "skip before fix the issue" )
60
+ self .skipTest ("skip before fix the issue" )
61
61
plugins .retrieval .enable = True
62
62
plugins .retrieval .args ["append" ] = True
63
63
plugins .retrieval .args ["input_path" ] = "../assets/docs/"
@@ -81,7 +81,7 @@ def test_retrieval_append(self):
81
81
plugins .retrieval .enable = False
82
82
83
83
def test_voice_chat (self ):
84
- self .skipTest (self , "skip before fix the issue" )
84
+ self .skipTest ("skip before fix the issue" )
85
85
plugins .tts .enable = True
86
86
plugins .tts .args ["output_audio_path" ] = "./response.wav"
87
87
pipeline_config = PipelineConfig (model_name_or_path = "facebook/opt-125m" ,
@@ -103,7 +103,7 @@ def test_quantization(self):
103
103
optimize_model (model = model , config = config )
104
104
105
105
def test_text_chat_stream (self ):
106
- self .skipTest (self , "skip before fix the issue" )
106
+ self .skipTest ("skip before fix the issue" )
107
107
config = PipelineConfig (model_name_or_path = "facebook/opt-125m" )
108
108
chatbot = build_chatbot (config )
109
109
stream_text = ""
0 commit comments