diff --git a/scripts/mock_client.py b/scripts/mock_client.py index d13b0c9..89b2b90 100644 --- a/scripts/mock_client.py +++ b/scripts/mock_client.py @@ -48,7 +48,8 @@ def build_message_bodies(analyse_container_type): message_dict.update({"targetFileExtension": "ORIGIN.pdf.gz", "responseFileExtension": "IMAGE_INFO.json.gz"}) if analyse_container_type == "ner": message_dict.update( - {"targetFileExtension": "TEXT.json.gz", "responseFileExtension": "NER_ENTITIES.json.gz"}) + {"targetFileExtension": "TEXT.json.gz", "responseFileExtension": "NER_ENTITIES.json.gz"} + ) return message_dict storage = get_s3_storage()