Skip to content
This repository was archived by the owner on Feb 3, 2025. It is now read-only.

Commit 1daf19e

Browse files
author
DEKHTIARJonathan
committed
[Benchmarking-Py] Adding pydot as a dependency of W&D
1 parent 5a6b859 commit 1daf19e

File tree

2 files changed

+5
-1
lines changed

2 files changed

+5
-1
lines changed

.gitignore

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -143,3 +143,6 @@ variables.data*
143143

144144
# Top Level Script Files
145145
/*.sh
146+
147+
# Benchmark Result Dir
148+
/benchmark_data

tftrt/benchmarking-python/nvidia_examples/wide_deep_tf2/run_inference.sh

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,8 @@ BASE_DIR="$( cd "$( dirname "${BASH_SOURCE[0]}" )" &> /dev/null && pwd )"
99
pip install --no-deps \
1010
tensorflow-transform~=0.24 \
1111
tensorflow-metadata~=0.24 \
12-
tfx-bsl~=0.24
12+
tfx-bsl~=0.24 \
13+
"pydot>=1.2,<2" `# necessary for tensorflow_transform`
1314

1415
python ${BASE_DIR}/infer.py \
1516
--data_dir=/data/outbrain \

0 commit comments

Comments
 (0)