From 080932be38ac791a29b6f3de922fa094bf58a7f3 Mon Sep 17 00:00:00 2001 From: "Shin, Eunwoo" Date: Wed, 24 Jul 2024 12:14:49 +0900 Subject: [PATCH] fix typo --- CHANGELOG.md | 2 +- src/otx/core/data/transform_libs/torchvision.py | 5 +++-- 2 files changed, 4 insertions(+), 3 deletions(-) diff --git a/CHANGELOG.md b/CHANGELOG.md index c0b981b2a4e..bb8fe6a455a 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -10,7 +10,7 @@ All notable changes to this project will be documented in this file. - Enable to override data configurations () -- Enable to use input_size at transforms in reipce +- Enable to use input_size at transforms in recipe () ### Bug fixes diff --git a/src/otx/core/data/transform_libs/torchvision.py b/src/otx/core/data/transform_libs/torchvision.py index 4fb37d97fa2..cd76aa53790 100644 --- a/src/otx/core/data/transform_libs/torchvision.py +++ b/src/otx/core/data/transform_libs/torchvision.py @@ -3122,9 +3122,10 @@ def generate(cls, config: SubsetConfig) -> Compose: if isinstance(config.transforms, Compose): return config.transforms + input_size = getattr(config, "input_size", None) transforms = [] for cfg_transform in config.transforms: - if (input_size := getattr(config, "input_size", None)) is not None: + if input_size is not None: cls._configure_input_size(cfg_transform, input_size) transform = cls._dispatch_transform(cfg_transform) transforms.append(transform) @@ -3164,7 +3165,7 @@ def check_type(value: Any, expected_type: Any) -> bool: # noqa: ANN401 available_types = typing.get_type_hints(model_cls.__init__).get(key) if available_types is None or check_type(_input_size, available_types): # pass tuple[int, int] cfg_transform["init_args"][key] = cls._eval_input_size_str( - val.replace("$(input_size)", f"({','.join(str(val) for val in _input_size)})"), + val.replace("$(input_size)", str(_input_size)), ) elif check_type(_input_size[0], available_types): # pass int cfg_transform["init_args"][key] = cls._eval_input_size_str(val.replace("$(input_size)", str(_input_size[0])))