From 93f994435111851f9aa062d90ca2c5054611df53 Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Wed, 14 Oct 2026 08:17:14 +0000
Subject: [PATCH 1/2] Declare the direct dependencies TRL gets transitively

---
 .github/workflows/tests.yml | 6 ++++++
 pyproject.toml              | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index d4e93bf1933..02760760fea 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -248,6 +248,12 @@ jobs:
             ".[dev]" \
             "accelerate==1.10.1" \
             "datasets==4.7.0" \
+            "huggingface_hub==0.34.0" \
+            "numpy==1.26.0" \
+            "pandas==2.1.1" \
+            "psutil==5.9.4" \
+            "pyarrow==21.0.0" \
+            "pyyaml==6.0.1" \
             "transformers==4.56.2"
 
       - name: Test with pytest
diff --git a/pyproject.toml b/pyproject.toml
index ca5e5b0b2a0..748de4e022b 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -30,8 +30,14 @@ requires-python = ">=3.11"
 dependencies = [
     "accelerate>=1.10.1",
     "datasets>=4.7.0",  # Support Json type and on_mixed_types="use_json"
+    "huggingface_hub>=0.34.0",
     "jinja2",
+    "numpy>=1.26.0",
     "packaging>20.0",
+    "pandas>=2.1.1",
+    "psutil>=5.9.4",
+    "pyarrow>=21.0.0",
+    "pyyaml>=6.0.1",
     "transformers>=4.56.2",
 ]
 dynamic = ["version"]

From a4faf33a155f9d6055729eabe835786c12c2cedc Mon Sep 17 00:00:00 2001
From: Albert Villanova del Moral
 <8515462+albertvillanova@users.noreply.github.com>
Date: Thu, 15 Oct 2026 07:18:11 +0000
Subject: [PATCH 2/2] Remove the minimum-versions job pins

---
 .github/workflows/tests.yml | 6 ------
 1 file changed, 6 deletions(-)

diff --git a/.github/workflows/tests.yml b/.github/workflows/tests.yml
index 775b39832d5..84c1838d36f 100644
--- a/.github/workflows/tests.yml
+++ b/.github/workflows/tests.yml
@@ -248,12 +248,6 @@ jobs:
             ".[dev]" \
             "accelerate==1.10.1" \
             "datasets==4.7.0" \
-            "huggingface_hub==0.34.0" \
-            "numpy==1.26.0" \
-            "pandas==2.1.1" \
-            "psutil==5.9.4" \
-            "pyarrow==21.0.0" \
-            "pyyaml==6.0.1" \
             "transformers==4.56.2"
 
       - name: Test with pytest