Update test_fasttext.py

meedan · Sep 22, 2023 · 290a740 · 290a740
1 parent 59a8bd0
commit 290a740
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/lib/model/test_fasttext.py b/test/lib/model/test_fasttext.py
@@ -12,7 +12,7 @@ def setUp(self):
     def test_respond(self, mock_fasttext_load_model, mock_hf_hub_download):
         mock_hf_hub_download.return_value = 'mocked_path'
         mock_fasttext_load_model.return_value = self.mock_model
-        self.mock_model.predict.return_value = (['__label__eng_Latn'], [0.9])
+        self.mock_model.predict.return_value = (['__label__eng_Latn'], np.array([0.902323124]))
         model = FasttextModel()  # Now it uses mocked functions
         query = [schemas.Message(body=schemas.TextInput(id="123", callback_url="http://example.com/callback", text="Hello, how are you?"))]
         response = model.respond(query)