Fix

Signed-off-by: Kim, Vinnam <vinnam.kim@intel.com>

docker/Dockerfile.cuda

@@ -50,7 +50,6 @@ RUN pip install --user --no-cache-dir --require-hashes --no-deps -r /tmp/require
 
 FROM install_packages AS download_pretrained_weights
 
-USER non-root
 WORKDIR ${NON_ROOT_HOME}
 
 COPY docker/download_pretrained_weights.py download_pretrained_weights.py
@@ -76,6 +75,3 @@ FROM cuda AS cuda_pretrained_ready
 ARG TORCH_CACHE=/home/non-root/.cache/torch
 
 COPY --chown=10001 --from=download_pretrained_weights ${TORCH_CACHE} ${TORCH_CACHE}
-
-USER non-root
-WORKDIR ${NON_ROOT_HOME}

src/otx/algo/modules/base_module.py

@@ -157,7 +157,7 @@ def _dump_init_info(self) -> None:
         with_file_handler = False
         # dump the information to the logger file if there is a `FileHandler`
         for handler in logger.handlers:
-            if isinstance(handler, FileHandler):
+            if isinstance(handler, FileHandler) and handler.stream is not None:
                 handler.stream.write("Name of parameter - Initialization information\n")
                 for name, param in self.named_parameters():
                     handler.stream.write(

src/otx/engine/hpo/hpo_api.py

@@ -70,7 +70,7 @@ def execute_hpo(
     engine.model.patch_optimizer_and_scheduler_for_hpo()
 
     hpo_workdir = Path(engine.work_dir) / "hpo"
-    hpo_workdir.mkdir(exist_ok=True)
+    hpo_workdir.mkdir(parents=True, exist_ok=True)
     hpo_configurator = HPOConfigurator(
         engine=engine,
         max_epochs=max_epochs,

src/otx/hpo/search_space.py

@@ -14,7 +14,13 @@
 
 logger = logging.getLogger()
 
-AVAILABLE_SEARCH_SPACE_TYPE = ["uniform", "quniform", "loguniform", "qloguniform", "choice"]
+AVAILABLE_SEARCH_SPACE_TYPE = [
+    "uniform",
+    "quniform",
+    "loguniform",
+    "qloguniform",
+    "choice",
+]
 
 
 class SingleSearchSpace:
@@ -58,7 +64,9 @@ def __init__(
         self._check_all_value_is_right()
 
     @property
-    def type(self) -> Literal["uniform", "loguniform", "quniform", "qloguniform", "choice"]:  # noqa: A003
+    def type(  # noqa: A003
+        self,
+    ) -> Literal["uniform", "loguniform", "quniform", "qloguniform", "choice"]:
         """Type of hyper parameter in search space."""
         return self._type
 
@@ -309,8 +317,13 @@ def __init__(
     ) -> None:
         self.search_space: dict[str, SingleSearchSpace] = {}
 
-        for key, val in search_space.items():  # pylint: disable=too-many-nested-blocks
-            self.search_space[key] = SingleSearchSpace(**val)
+        try:
+            for key, val in search_space.items():  # pylint: disable=too-many-nested-blocks
+                self.search_space[key] = SingleSearchSpace(**val)
+        except Exception:
+            msg = f"Failed to create SingleSearchSpace. key={key}, val={val}"
+            logging.exception(msg)
+            raise
 
     def __getitem__(self, key: str) -> SingleSearchSpace:
         """Get search space by key."""