From 549ccc6c7b0a41eda1f9e507a27edc5da35088a4 Mon Sep 17 00:00:00 2001
From: Arthur Zucker <arthur.zucker@gmail.com>
Date: Wed, 22 May 2024 07:57:30 +0200
Subject: [PATCH] [build-ci-image]

---
 setup.py               | 2 +-
 utils/tests_fetcher.py | 1 +
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 15226081ed2096..3d6c78fd9a695d 100644
--- a/setup.py
+++ b/setup.py
@@ -173,7 +173,7 @@
     "tensorflow-cpu>2.9,<2.16",
     "tensorflow>2.9,<2.16",
     "tensorflow-text<2.16",
-    "tensorflow-probability<2.16",
+    "tensorflow-probability<0.24",
     "tf2onnx",
     "timeout-decorator",
     "timm<=0.9.16",
diff --git a/utils/tests_fetcher.py b/utils/tests_fetcher.py
index 7a44ed82033e34..c75479757bca81 100644
--- a/utils/tests_fetcher.py
+++ b/utils/tests_fetcher.py
@@ -93,6 +93,7 @@
     "vit",
     "whisper",
     # Pipeline-specific model (to be sure each pipeline has one model in this list)
+    "tapas",
     "vilt",
     "clap",
     "detr",