From d1cc61a78b8c502d6d735e9430db06fbbe1c72cf Mon Sep 17 00:00:00 2001 From: Davanum Srinivas Date: Tue, 20 Jan 2026 12:09:49 -0500 Subject: [PATCH] test/images: fix tf-wide-deep build by using models v1.9.0 The wide_deep directory was removed from tensorflow/models after v1.9.0. Commit f8ca585b666 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 --- test/images/node-perf/tf-wide-deep/Dockerfile | 15 +++++++++------ test/images/node-perf/tf-wide-deep/VERSION | 2 +- 2 files changed, 10 insertions(+), 7 deletions(-) diff --git a/test/images/node-perf/tf-wide-deep/Dockerfile b/test/images/node-perf/tf-wide-deep/Dockerfile index 6ed7f1fa304..6ab4dc61436 100644 --- a/test/images/node-perf/tf-wide-deep/Dockerfile +++ b/test/images/node-perf/tf-wide-deep/Dockerfile @@ -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 diff --git a/test/images/node-perf/tf-wide-deep/VERSION b/test/images/node-perf/tf-wide-deep/VERSION index c068b2447cc..c239c60cba2 100644 --- a/test/images/node-perf/tf-wide-deep/VERSION +++ b/test/images/node-perf/tf-wide-deep/VERSION @@ -1 +1 @@ -1.4 +1.5