github/kubernetes
1
0
Fork 0
mirror of https://github.com/k3s-io/kubernetes.git synced 2026-07-18 04:54:54 +00:00
Code Issues Projects Releases Wiki Activity

test/images: fix tf-wide-deep build by using models v1.9.0

Browse Source 
The wide_deep directory was removed from tensorflow/models after v1.9.0.
Commit f8ca585b66 updated models from v1.9.0 to v2.15.0, breaking the
build because /models-2.15.0/official/wide_deep/data_download.py no
longer exists.

Revert to models v1.9.0 which contains the wide_deep implementation.
The v1.9.0 code uses tf.estimator APIs which are still available in
TensorFlow 2.x.

Fixes: https://prow.k8s.io/view/gs/kubernetes-ci-logs/logs/post-kubernetes-push-e2e-node-perf-tf-wide-deep-test-images/2011957382287986688

Signed-off-by: Davanum Srinivas <davanum@gmail.com>
This commit is contained in:
Davanum Srinivas
2026-01-20 12:09:49 -05:00
parent cb077823fb
commit d1cc61a78b
2 changed files with 10 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
test/images/node-perf/tf-wide-deep/Dockerfile
View File

@@ -21,13 +21,16 @@ RUN apt-get update && apt-get install -y wget time
RUN pip install tensorflow==2.20.0
RUN wget https://github.com/tensorflow/models/archive/v2.15.0.tar.gz \
&& tar xzf v2.15.0.tar.gz \
&& rm -f v2.15.0.tar.gz
# Use models v1.9.0 which contains the wide_deep implementation.
# The wide_deep directory was removed in later versions of tensorflow/models.
# The v1.9.0 code uses tf.estimator APIs which are still available in TF 2.x.
RUN wget https://github.com/tensorflow/models/archive/v1.9.0.tar.gz \
&& tar xzf v1.9.0.tar.gz \
&& rm -f v1.9.0.tar.gz
RUN python /models-2.15.0/official/wide_deep/data_download.py
RUN python /models-1.9.0/official/wide_deep/data_download.py
WORKDIR $HOME/models-2.15.0/official/wide_deep
ENV PYTHONPATH=$PYTHONPATH:$HOME/models-2.15.0
WORKDIR $HOME/models-1.9.0/official/wide_deep
ENV PYTHONPATH=$PYTHONPATH:$HOME/models-1.9.0
ENTRYPOINT python ./wide_deep.py

2
test/images/node-perf/tf-wide-deep/VERSION
View File

@@ -1 +1 @@
1.4
1.5