Fix pythonpath in rust ci

nhynes · Oct 7, 2018 · 744bfbd · 744bfbd
1 parent 275e200
commit 744bfbd
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/tests/scripts/task_rust.sh b/tests/scripts/task_rust.sh
@@ -1,9 +1,12 @@
 #!/bin/bash
-export LD_LIBRARY_PATH=lib:${LD_LIBRARY_PATH}
-export PYTHONPATH=python:nnvm/python:topi/python
 
 set -e
 
+export LD_LIBRARY_PATH=lib:$LD_LIBRARY_PATH
+
+tvm_root="$(git rev-parse --show-toplevel)"
+export PYTHONPATH="$tvm_root/python":"$tvm_root/nnvm/python":"$tvm_root/topi/python"
+
 cd rust
 cargo fmt -- --check