Properly add the fucntion to fix holoscan init file.

Signed-off-by: M Q <mingmelvinq@nvidia.com>

Author: MMelQin

monai/deploy/utils/importutil.py

@@ -341,42 +341,6 @@ def dist_requires(project_name: str) -> List[str]:
     return []
 
 
-if __name__ == "__main__":
-    """Utility functions that can be used in the command line."""
-
-    argv = sys.argv
-    if len(argv) == 3 and argv[1] == "is_dist_editable":
-        if is_dist_editable(argv[2]):
-            sys.exit(0)
-        else:
-            sys.exit(1)
-    if len(argv) == 3 and argv[1] == "dist_module_path":
-        module_path = dist_module_path(argv[2])
-        if module_path:
-            print(module_path)
-            sys.exit(0)
-        else:
-            sys.exit(1)
-    if len(argv) == 3 and argv[1] == "is_module_installed":
-        is_installed = is_module_installed(argv[2])
-        if is_installed:
-            sys.exit(0)
-        else:
-            sys.exit(1)
-    if len(argv) == 3 and argv[1] == "dist_requires":
-        requires = dist_requires(argv[2])
-        if requires:
-            print("\n".join(requires))
-            sys.exit(0)
-        else:
-            sys.exit(1)
-    if len(argv) >= 3 and argv[1] == "get_package_info":
-        import json
-
-        app = get_application(argv[2])
-        if app:
-            print(json.dumps(app.get_package_info(argv[3] if len(argv) > 3 else ""), indent=2))
-
 holoscan_init_content_txt = """
 # SPDX-FileCopyrightText: Copyright (c) 2022-2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 # SPDX-License-Identifier: Apache-2.0
@@ -435,9 +399,56 @@ def __getattr__(name):
 def fix_holoscan_import():
     """Fix holoscan __init__ to enable lazy load for avoiding failure on loading low level libs."""
 
-    holoscan_package_name = "holoscan"
-    holoscan_package_path = dist_module_path(holoscan_package_name)
-    holoscan_core_init_path = Path(holoscan_package_path) / holoscan_package_name / "__init__.py"
+    try:
+        project_name = "holoscan"
+        holoscan_init_path = Path(dist_module_path(project_name)) / project_name / "__ini__.py"
+
+        with open(str(holoscan_init_path), "w") as f_w:
+            f_w.write(holoscan_init_content_txt)
+        return str(holoscan_init_path)
+    except Exception as ex:
+        return ex
+
 
-    with open(holoscan_core_init_path, "w") as f_w:
-        f_w.write(holoscan_init_content_txt)
+if __name__ == "__main__":
+    """Utility functions that can be used in the command line."""
+
+    argv = sys.argv
+    if len(argv) == 2 and argv[1] == "fix_holoscan_import":
+        file_path = fix_holoscan_import()
+        if file_path:
+            print(file_path)
+            sys.exit(0)
+        else:
+            sys.exit(1)
+    if len(argv) == 3 and argv[1] == "is_dist_editable":
+        if is_dist_editable(argv[2]):
+            sys.exit(0)
+        else:
+            sys.exit(1)
+    if len(argv) == 3 and argv[1] == "dist_module_path":
+        module_path = dist_module_path(argv[2])
+        if module_path:
+            print(module_path)
+            sys.exit(0)
+        else:
+            sys.exit(1)
+    if len(argv) == 3 and argv[1] == "is_module_installed":
+        is_installed = is_module_installed(argv[2])
+        if is_installed:
+            sys.exit(0)
+        else:
+            sys.exit(1)
+    if len(argv) == 3 and argv[1] == "dist_requires":
+        requires = dist_requires(argv[2])
+        if requires:
+            print("\n".join(requires))
+            sys.exit(0)
+        else:
+            sys.exit(1)
+    if len(argv) >= 3 and argv[1] == "get_package_info":
+        import json
+
+        app = get_application(argv[2])
+        if app:
+            print(json.dumps(app.get_package_info(argv[3] if len(argv) > 3 else ""), indent=2))

run

@@ -325,18 +325,17 @@ install_python_dev_deps() {
         exit 1
     fi
 
-    install_edit_mode
-
     # Adding temp fix to address the issue of holoscan sdk dragging in low level dependencies, e.g. libcuda.so
     fix_holoscan_import
+
+    install_edit_mode
 }
 
 fix_holoscan_import() {
     local holoscan_package_path=$(${MONAI_PY_EXE} $TOP/monai/deploy/utils/importutil.py dist_module_path holoscan)
     c_echo b "holoscan_package_path : " Z "${holoscan_package_path}"
-    c_echo b "fixing holoscan package module import..."
-    ${MONAI_PY_EXE} $TOP/monai/deploy/utils/importutil.py fix_holoscan_import
-    c_echo b "done fixing."
+    init_file_path=$(${MONAI_PY_EXE} $TOP/monai/deploy/utils/importutil.py fix_holoscan_import)
+    c_echo b "done fixing file: " Z "${init_file_path}"
 }
 
 install_edit_mode() {