change input_size type to Any to solve omegaconf issue

openvinotoolkit · Jul 24, 2024 · 096f3d1 · 096f3d1
1 parent 080932b
commit 096f3d1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/otx/core/config/data.py b/src/otx/core/config/data.py
@@ -60,7 +60,9 @@ class SubsetConfig:
     num_workers: int = 2
     sampler: SamplerConfig = field(default_factory=lambda: SamplerConfig())
     to_tv_image: bool = True
-    input_size: int | tuple[int, int] | None = None
+    input_size: (
+        Any
+    ) = None  # type is `int | tuple[int, int] | None` TODO (eunwoosh): Revisit after error above is solved
 
 
 @dataclass

diff --git a/src/otx/core/data/transform_libs/torchvision.py b/src/otx/core/data/transform_libs/torchvision.py
@@ -3168,7 +3168,9 @@ def check_type(value: Any, expected_type: Any) -> bool:  # noqa: ANN401
                     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])))
+                cfg_transform["init_args"][key] = cls._eval_input_size_str(
+                    val.replace("$(input_size)", str(_input_size[0])),
+                )
             else:
                 msg = f"{key} argument should be able to get int or tuple[int, int], but it can get {available_types}"
                 raise RuntimeError(msg)