Bring back tools/torchscript_e2e_test.sh

Accidentally deleted it in the last commit.
2021-09-23 19:01:23 +00:00 · 2021-09-23 19:01:23 +00:00 · e291aa688f
parent 81a9cf7800
commit e291aa688f
1 changed files with 12 additions and 0 deletions
--- a/tools/torchscript_e2e_test.sh
+++ b/tools/torchscript_e2e_test.sh
@ -0,0 +1,12 @@
+#!/bin/bash
+set -euo pipefail
+
+src_dir="$(realpath $(dirname $0)/..)"
+
+cd "$src_dir"
+
+# Ensure PYTHONPATH is set for export to child processes, even if empty.
+export PYTHONPATH=${PYTHONPATH-}
+source .env
+
+python -m e2e_testing.torchscript.main "$@"