[Benchmarking-Py] Adding pydot as a dependency of W&D

tensorflow · Jul 19, 2022 · 1daf19e · 1daf19e
1 parent 5a6b859
commit 1daf19e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/.gitignore b/.gitignore
@@ -143,3 +143,6 @@ variables.data*
 
 # Top Level Script Files
 /*.sh
+
+# Benchmark Result Dir
+/benchmark_data
diff --git a/tftrt/benchmarking-python/nvidia_examples/wide_deep_tf2/run_inference.sh b/tftrt/benchmarking-python/nvidia_examples/wide_deep_tf2/run_inference.sh
@@ -9,7 +9,8 @@ BASE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
 pip install --no-deps \
     tensorflow-transform~=0.24 \
     tensorflow-metadata~=0.24 \
-    tfx-bsl~=0.24
+    tfx-bsl~=0.24 \
+    "pydot>=1.2,<2"  `# necessary for tensorflow_transform`
 
 python ${BASE_DIR}/infer.py \
     --data_dir=/data/outbrain \