From c881644133775d020cffd83cdd62c8c83fba2484 Mon Sep 17 00:00:00 2001 From: Wojtek Kowaluk Date: Sat, 20 May 2023 16:23:00 +0200 Subject: [PATCH] Do not attampt to serialize Images from params for metadata --- src/utils/file_system.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/src/utils/file_system.py b/src/utils/file_system.py index 805d309..5d05196 100644 --- a/src/utils/file_system.py +++ b/src/utils/file_system.py @@ -7,7 +7,8 @@ def create_png_info(metadata): png_info = PngImagePlugin.PngInfo() - png_info.add_text("kubin_image_metadata", metadata) + if metadata: + png_info.add_text("kubin_image_metadata", metadata) return png_info def create_filename(path, params): @@ -34,8 +35,11 @@ def create_filename(path, params): def save_output(output_dir, task_type, images, params=None): output = [] + params_as_json = None - params_as_json = None if params is None else json.dumps(params, skipkeys=True) + if params: + params = {v: k for k, v in params.items() if not isinstance(v, Image.Image)} + params_as_json = json.dumps(params, skipkeys=True) for img in images: path = f'{output_dir}/{task_type}'