From c9f9cd165b6a69c69c97e3750c7cd1dc04853611 Mon Sep 17 00:00:00 2001
From: Nicolas Patry <patry.nicolas@protonmail.com>
Date: Mon, 20 Jan 2025 16:27:39 +0100
Subject: [PATCH] Trying to avoid the random timeout.

---
 .github/workflows/build.yaml | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml
index c43d8eb9..83690559 100644
--- a/.github/workflows/build.yaml
+++ b/.github/workflows/build.yaml
@@ -214,4 +214,6 @@ jobs:
           export HF_TOKEN=${{ secrets.HF_TOKEN }}
           echo $DOCKER_IMAGE
           docker pull $DOCKER_IMAGE
+          # Run once only the bloom tests to try and avoid the random timeout.
+          pytest -s -vv --release integration-tests/models/test_bloom_560m.py
           pytest -s -vv integration-tests ${PYTEST_FLAGS} ${EXTRA_PYTEST}