Minimum requirement for pyyaml

bigscience-workshop · Jul 12, 2021 · ad42054 · ad42054
1 parent fb5665b
commit ad42054
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git 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
@@ -119,7 +119,7 @@
     "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
@@ -36,7 +36,7 @@
     "parameterized": "parameterized",
     "protobuf": "protobuf",
     "psutil": "psutil",
-    "pyyaml": "pyyaml",
+    "pyyaml": "pyyaml>=5.1",
     "pydantic": "pydantic",
     "pytest": "pytest",
     "pytest-sugar": "pytest-sugar",