fix(ci): fix flaky docker build behavior in ci

.github/workflows/test-pytorch-xla-tpu-tgi-integration.yml

@@ -1,6 +1,7 @@
 name: Optimum TPU / Test TGI on TPU / Integration Tests
 
 on:
+  push:
   schedule:
     - cron: '0 4 * * *' # run at 4 AM UTC
     # This can be used to allow manually triggering nightlies from the web interface
@@ -32,9 +33,10 @@ jobs:
           sudo apt-get install -y python3 python3-pip
           sudo ln -s /usr/bin/python3 /usr/bin/python
 
+      # To build the docker image in the ci, we need to use the network host option
       - name: Build TGI Docker Image
         run: |
-          make tpu-tgi
+          make tpu-tgi NETWORK=host
 
       - name: Run integration tests
         run: |

Makefile

@@ -42,11 +42,14 @@ clean:
 	rm -rf dist deps
 	make -C text-generation-inference/server/ clean
 
+# normal usage: make tpu-tgi
+# ci usage: make tpu-tgi NETWORK=host, to build the docker image with the network host option
 tpu-tgi:
 	docker build --rm -f text-generation-inference/docker/Dockerfile \
 	             --build-arg VERSION=$(VERSION) \
 	             --build-arg TGI_VERSION=$(TGI_VERSION) \
 	             --ulimit nofile=100000:100000 \
+				 $(if $(NETWORK),--network $(NETWORK),) \
 	             -t huggingface/optimum-tpu:$(VERSION)-tgi .
 	docker tag huggingface/optimum-tpu:$(VERSION)-tgi huggingface/optimum-tpu:latest