Skip to content

Commit 20c5ec2

Browse files
lint fix
1 parent bf61416 commit 20c5ec2

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

toolium/test/utils/ai_utils/test_text_similarity.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -204,7 +204,8 @@ def test_assert_text_similarity_with_configured_and_explicit_method_and_model(si
204204

205205
input_text = 'Today it will be sunny'
206206
expected_text = 'Today is sunny'
207-
assert_text_similarity(input_text, expected_text, threshold=0.8, similarity_method='spacy', model_name='en_core_web_lg')
207+
assert_text_similarity(input_text, expected_text, threshold=0.8, similarity_method='spacy',
208+
model_name='en_core_web_lg')
208209
similarity_mock.assert_called_once_with(input_text, expected_text, model_name='en_core_web_lg')
209210

210211

@@ -220,7 +221,8 @@ def test_assert_text_similarity_with_explicit_openai(similarity_mock):
220221

221222
input_text = 'Today it will be sunny'
222223
expected_text = 'Today is sunny'
223-
assert_text_similarity(input_text, expected_text, threshold=0.8, similarity_method='openai', azure=True, model_name='gpt-4o-mini')
224+
assert_text_similarity(input_text, expected_text, threshold=0.8, similarity_method='openai',
225+
azure=True, model_name='gpt-4o-mini')
224226
similarity_mock.assert_called_once_with(input_text, expected_text, azure=True, model_name='gpt-4o-mini')
225227

226228

@@ -237,4 +239,4 @@ def test_azure_openai_request_params(similarity_mock):
237239
input_text = 'Today it will be sunny'
238240
expected_text = 'Today is sunny'
239241
assert_text_similarity(input_text, expected_text, threshold=0.8)
240-
similarity_mock.assert_called_once_with(input_text, expected_text, azure=True)
242+
similarity_mock.assert_called_once_with(input_text, expected_text, azure=True)

0 commit comments

Comments
 (0)