From ad42054278f407ae92495580834aa993cf7f4028 Mon Sep 17 00:00:00 2001
From: Sylvain Gugger <sylvain.gugger@gmail.com>
Date: Mon, 12 Jul 2021 09:55:36 -0400
Subject: [PATCH] Minimum requirement for pyyaml

---
 .github/conda/meta.yaml                       | 4 ++--
 setup.py                                      | 2 +-
 src/transformers/dependency_versions_table.py | 2 +-
 3 files changed, 4 insertions(+), 4 deletions(-)

diff --git a/.github/conda/meta.yaml b/.github/conda/meta.yaml
index b18af06c3a1..248fbcc3726 100644
--- a/.github/conda/meta.yaml
+++ b/.github/conda/meta.yaml
@@ -26,7 +26,7 @@ requirements:
     - regex !=2019.12.17
     - protobuf
     - tokenizers >=0.10.1,<0.11.0
-    - pyyaml
+    - pyyaml >= 5.1
   run:
     - python
     - numpy >=1.17
@@ -41,7 +41,7 @@ requirements:
     - regex !=2019.12.17
     - protobuf
     - tokenizers >=0.10.1,<0.11.0
-    - pyyaml
+    - pyyaml >= 5.1
 
 test:
   imports:
diff --git a/setup.py b/setup.py
index 7b0eb6172b8..7be7700a56f 100644
--- a/setup.py
+++ b/setup.py
@@ -119,7 +119,7 @@ _deps = [
     "parameterized",
     "protobuf",
     "psutil",
-    "pyyaml",
+    "pyyaml>=5.1",
     "pydantic",
     "pytest",
     "pytest-sugar",
diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py
index 7bea4c79cb6..942a3faf99c 100644
--- a/src/transformers/dependency_versions_table.py
+++ b/src/transformers/dependency_versions_table.py
@@ -36,7 +36,7 @@ deps = {
     "parameterized": "parameterized",
     "protobuf": "protobuf",
     "psutil": "psutil",
-    "pyyaml": "pyyaml",
+    "pyyaml": "pyyaml>=5.1",
     "pydantic": "pydantic",
     "pytest": "pytest",
     "pytest-sugar": "pytest-sugar",