[discovery] Fix linter errors on test code
@@ -559,8 +559,7 @@ def test_add_training_data_query():
collection_id=collection_id,
natural_language_query=natural_language_query,
filter=filter,
- examples=examples
- )
+ examples=examples)
assert response == mock_response
@@ -643,8 +642,7 @@ def test_list_training_data_query_examples():
response = service.list_training_data_query_examples(
environment_id=environment_id,
collection_id=collection_id,
- query_id=query_id
- )
+ query_id=query_id)
assert response == mock_response
@@ -679,14 +677,13 @@ def test_add_training_data_query_example():
query_id=query_id,
document_id=document_id,
relevance=relevance,
- cross_reference=cross_reference
- )
+ cross_reference=cross_reference)
assert response == mock_response
@responses.activate
-def test_delete_training_data_query():
+def test_delete_training_data_query_example():
examples_endpoint = '/v1/environments/{0}/collections/{1}/training_data' + \
'/{2}/examples/{3}'
query_id = 'queryid'
@@ -705,8 +702,7 @@ def test_delete_training_data_query():
environment_id=environment_id,
collection_id=collection_id,
query_id=query_id,
- example_id=example_id
- )
+ example_id=example_id)
assert response.status_code == 204
@@ -740,8 +736,7 @@ def test_get_training_data_query_example():
environment_id=environment_id,
collection_id=collection_id,
query_id=query_id,
- example_id=example_id
- )
+ example_id=example_id)
assert response == mock_response
@@ -779,7 +774,6 @@ def test_update_training_data_query_example():
query_id=query_id,
example_id=example_id,
relevance=relevance,
- cross_reference=cross_reference
- )
+ cross_reference=cross_reference)
assert response == mock_response
