From b7bd0e389884314122e160a300e4c75ebb26d06d Mon Sep 17 00:00:00 2001
From: Ricardo Galli <gallir@gmail.com>
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