From 9761db73b0fe3eefbc2fa4edbec4eb8004e1f2bf Mon Sep 17 00:00:00 2001 From: Mark Date: Tue, 14 Nov 2023 17:06:23 +0000 Subject: [PATCH] Updating indents. --- cli/text2embeddings.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/cli/text2embeddings.py b/cli/text2embeddings.py index e1b997b..3d9bd85 100644 --- a/cli/text2embeddings.py +++ b/cli/text2embeddings.py @@ -178,9 +178,11 @@ def run_as_cli( task_output_path = os.path.join(output_dir, task.document_id + ".json") try: - write_json_to_s3(task.model_dump_json(), task_output_path) if s3 else Path( - task_output_path - ).write_text(task.model_dump_json()) + write_json_to_s3( + task.model_dump_json(indent=2), task_output_path + ) if s3 else Path(task_output_path).write_text( + task.model_dump_json(indent=2) + ) except Exception as e: logger.info( "Failed to write embeddings data to s3.",