From 6a9987a233842b8b3f5fc312f6567f2b8b18e029 Mon Sep 17 00:00:00 2001 From: galipremsagar Date: Tue, 28 Sep 2021 07:08:48 -0700 Subject: [PATCH] pin transformers --- conda/environments/cudf_dev_cuda11.0.yml | 2 +- conda/environments/cudf_dev_cuda11.2.yml | 2 +- python/cudf/requirements/cuda-11.0/dev_requirements.txt | 2 +- python/cudf/requirements/cuda-11.2/dev_requirements.txt | 2 +- python/cudf/setup.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/conda/environments/cudf_dev_cuda11.0.yml b/conda/environments/cudf_dev_cuda11.0.yml index 1476c294682..9723810c7b2 100644 --- a/conda/environments/cudf_dev_cuda11.0.yml +++ b/conda/environments/cudf_dev_cuda11.0.yml @@ -55,7 +55,7 @@ dependencies: - protobuf - nvtx>=0.2.1 - cachetools - - transformers + - transformers<=4.10.3 - pydata-sphinx-theme - pip: - git+https://github.com/dask/dask.git@2021.09.1 diff --git a/conda/environments/cudf_dev_cuda11.2.yml b/conda/environments/cudf_dev_cuda11.2.yml index 37f1899fcf5..6a8203ffc1c 100644 --- a/conda/environments/cudf_dev_cuda11.2.yml +++ b/conda/environments/cudf_dev_cuda11.2.yml @@ -55,7 +55,7 @@ dependencies: - protobuf - nvtx>=0.2.1 - cachetools - - transformers + - transformers<=4.10.3 - pydata-sphinx-theme - pip: - git+https://github.com/dask/dask.git@2021.09.1 diff --git a/python/cudf/requirements/cuda-11.0/dev_requirements.txt b/python/cudf/requirements/cuda-11.0/dev_requirements.txt index b76011ca2ab..e68278905ed 100644 --- a/python/cudf/requirements/cuda-11.0/dev_requirements.txt +++ b/python/cudf/requirements/cuda-11.0/dev_requirements.txt @@ -35,6 +35,6 @@ sphinx sphinx-copybutton sphinx-markdown-tables sphinxcontrib-websupport -transformers +transformers<=4.10.3 typing_extensions wheel diff --git a/python/cudf/requirements/cuda-11.2/dev_requirements.txt b/python/cudf/requirements/cuda-11.2/dev_requirements.txt index 146c2c9781a..b87987bb21d 100644 --- a/python/cudf/requirements/cuda-11.2/dev_requirements.txt +++ b/python/cudf/requirements/cuda-11.2/dev_requirements.txt @@ -35,6 +35,6 @@ sphinx sphinx-copybutton sphinx-markdown-tables sphinxcontrib-websupport -transformers +transformers<=4.10.3 typing_extensions wheel diff --git a/python/cudf/setup.py b/python/cudf/setup.py index 6d210eff071..abf632559cc 100644 --- a/python/cudf/setup.py +++ b/python/cudf/setup.py @@ -51,7 +51,7 @@ "hypothesis" "mimesis", "pyorc", "msgpack", - "transformers", + "transformers<=4.10.3", ] }