From 3fd7de49f4d53832621ef67cfd1825b000323250 Mon Sep 17 00:00:00 2001 From: Sylvain Gugger <35901082+sgugger@users.noreply.github.com> Date: Fri, 20 May 2022 11:00:58 -0400 Subject: [PATCH] Pin dill to fix examples (#17368) * Pin dill for now * Try this version? * force install * Actually use dep in testing * Try a larger pin --- setup.py | 2 ++ src/transformers/dependency_versions_table.py | 1 + 2 files changed, 3 insertions(+) diff --git a/setup.py b/setup.py index a7e11799f5..62b5fa6864 100644 --- a/setup.py +++ b/setup.py @@ -104,6 +104,7 @@ _deps = [ "dataclasses", "datasets", "deepspeed>=0.6.4", + "dill<0.3.5", "fairscale>0.3", "faiss-cpu", "fastapi", @@ -284,6 +285,7 @@ extras["testing"] = ( "parameterized", "psutil", "datasets", + "dill", "pytest-timeout", "black", "sacrebleu", diff --git a/src/transformers/dependency_versions_table.py b/src/transformers/dependency_versions_table.py index 4e6541771f..6db2a5feb2 100644 --- a/src/transformers/dependency_versions_table.py +++ b/src/transformers/dependency_versions_table.py @@ -10,6 +10,7 @@ deps = { "dataclasses": "dataclasses", "datasets": "datasets", "deepspeed": "deepspeed>=0.6.4", + "dill": "dill<0.3.5", "fairscale": "fairscale>0.3", "faiss-cpu": "faiss-cpu", "fastapi": "fastapi",