feat(//py): Catch when bazel is not in path and error out when running

setup.py Signed-off-by: Naren Dasan <naren@narendasan.com> Signed-off-by: Naren Dasan <narens@nvidia.com>
pytorch · May 13, 2021 · 1da999d · 1da999d
1 parent 1fb31bc
commit 1da999d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/py/setup.py b/py/setup.py
@@ -46,6 +46,8 @@ def is_exe(fpath):
 
 BAZEL_EXE = which("bazel")
 
+if BAZEL_EXE is None:
+    sys.exit("Could not find bazel in PATH")
 
 def build_libtrtorch_pre_cxx11_abi(develop=True, use_dist_dir=True, cxx11_abi=False):
     cmd = [BAZEL_EXE, "build"]