Adding back sys.path

microsoft · Jan 18, 2024 · 10e2aea · 10e2aea
1 parent 8da3a84
commit 10e2aea
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tools/ci_build/build.py b/tools/ci_build/build.py
@@ -13,10 +13,14 @@
 import subprocess
 import sys
 from pathlib import Path
-from ..python.util import android as android
 
+SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
+REPO_DIR = os.path.normpath(os.path.join(SCRIPT_DIR, "..", ".."))
+
+sys.path.insert(0, os.path.join(REPO_DIR, "tools", "python"))
 # The Following utility packages are from onnxruntime/tools/python/util
-from ..python.util import (
+import util.android as android  # noqa: E402
+from util import (
     get_logger,
     is_linux,
     is_macOS,
@@ -42,7 +46,6 @@
     setup_cann_vars,
 )  # noqa: E402
 
-SCRIPT_DIR = os.path.dirname(os.path.realpath(__file__))
 log = get_logger("build")
 check_python_version()