From b7bd0e389884314122e160a300e4c75ebb26d06d Mon Sep 17 00:00:00 2001 From: Ricardo Galli Date: Tue, 21 Jan 2025 00:40:34 +0100 Subject: [PATCH] Fix ignoring git branch, use git clone instead of curl --- tensorflow_serving/tools/docker/Dockerfile.devel | 4 +++- tensorflow_serving/tools/docker/Dockerfile.devel-gpu | 4 +++- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/tensorflow_serving/tools/docker/Dockerfile.devel b/tensorflow_serving/tools/docker/Dockerfile.devel index fe28614b718..ae30a6efe6b 100644 --- a/tensorflow_serving/tools/docker/Dockerfile.devel +++ b/tensorflow_serving/tools/docker/Dockerfile.devel @@ -93,7 +93,9 @@ RUN mkdir /bazel && \ # Download TF Serving sources (optionally at specific commit). WORKDIR /tensorflow-serving -RUN curl -sSL --retry 5 https://github.com/tensorflow/serving/tarball/${TF_SERVING_VERSION_GIT_COMMIT} | tar --strip-components=1 -xzf - +RUN git clone --single-branch --branch=${TF_SERVING_VERSION_GIT_BRANCH} https://github.com/tensorflow/serving /tensorflow-serving && \ + cd /tensorflow-serving && \ + git reset --hard ${TF_SERVING_VERSION_GIT_COMMIT} FROM base_build as binary_build # Build, and install TensorFlow Serving diff --git a/tensorflow_serving/tools/docker/Dockerfile.devel-gpu b/tensorflow_serving/tools/docker/Dockerfile.devel-gpu index 66032984cb2..dc037334065 100644 --- a/tensorflow_serving/tools/docker/Dockerfile.devel-gpu +++ b/tensorflow_serving/tools/docker/Dockerfile.devel-gpu @@ -30,7 +30,9 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ # Download TF Serving sources (optionally at specific commit). WORKDIR /tensorflow-serving -RUN curl -sSL --retry 5 https://github.com/tensorflow/serving/tarball/${TF_SERVING_VERSION_GIT_COMMIT} | tar --strip-components=1 -xzf - +RUN git clone --single-branch --branch=${TF_SERVING_VERSION_GIT_BRANCH} https://github.com/tensorflow/serving /tensorflow-serving && \ + cd /tensorflow-serving && \ + git reset --hard ${TF_SERVING_VERSION_GIT_COMMIT} RUN /tensorflow-serving/tensorflow_serving/tools/docker/setup.sources.sh