Fix python import path of other tools

iree-org · May 2, 2023 · 177a3da · 177a3da
1 parent eb885ed
commit 177a3da
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 2 deletions.
diff --git a/build_tools/benchmarks/diff_local_benchmarks.py b/build_tools/benchmarks/diff_local_benchmarks.py
@@ -11,8 +11,13 @@
                                    --target=/path/to/target_benchmarks.json
 """
 
-import argparse
 import pathlib
+import sys
+
+# Add build_tools python dir to the search path.
+sys.path.insert(0, str(pathlib.Path(__file__).parent.with_name("python")))
+
+import argparse
 
 from typing import Optional
 

diff --git a/build_tools/benchmarks/upload_benchmarks_to_dashboard.py b/build_tools/benchmarks/upload_benchmarks_to_dashboard.py
@@ -15,10 +15,15 @@
   python3 upload_benchmarks.py /path/to/benchmark/json/file
 """
 
+import pathlib
+import sys
+
+# Add build_tools python dir to the search path.
+sys.path.insert(0, str(pathlib.Path(__file__).parent.with_name("python")))
+
 import argparse
 import json
 import os
-import pathlib
 import requests
 
 from typing import Any, Dict, Optional, Union