From 0cf2cd71de72a812340953e6ad9aa69ad5823b41 Mon Sep 17 00:00:00 2001 From: Joel Holdbrooks Date: Mon, 23 Jan 2023 00:14:50 -0800 Subject: [PATCH] Fix mistake in ImageCaption.parse --- data/image_train_item.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/data/image_train_item.py b/data/image_train_item.py index 0e05145..db6f4e4 100644 --- a/data/image_train_item.py +++ b/data/image_train_item.py @@ -137,7 +137,7 @@ class ImageCaption: """ (file_name, _) = os.path.splitext(os.path.basename(file_path)) caption = file_name.split("_")[0] - return ImageCaption(caption, 1.0, [], [], DEFAULT_MAX_CAPTION_LENGTH, False) + return ImageCaption.parse(caption) @staticmethod def from_text_file(file_path: str, default_caption: OptionalImageCaption=None) -> OptionalImageCaption: