Daniël de Kok 2020-09-30 10:35:48 +02:00
parent e1773ee0bb
commit 71b3dbc799

View File

@ -17,13 +17,13 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "transformers"; pname = "transformers";
version = "3.2.0"; version = "3.3.1";
src = fetchFromGitHub { src = fetchFromGitHub {
owner = "huggingface"; owner = "huggingface";
repo = pname; repo = pname;
rev = "v${version}"; rev = "v${version}";
sha256 = "0jj94153kgdyklra30xcszxv11hwzfigzy82fgvgzvbwlxv3a1j5"; sha256 = "1j9nzhl0zw5z9rnvzfih7v6bax353rxp05b3f0cvkii3b5dbkc2j";
}; };
propagatedBuildInputs = [ propagatedBuildInputs = [
@ -52,9 +52,12 @@ buildPythonPackage rec {
preCheck = '' preCheck = ''
export HOME="$TMPDIR" export HOME="$TMPDIR"
# This test requires the nlp module, which we haven't # This test requires the `datasets` module to download test
# packaged yet. However, nlp is optional for transformers # data. However, since we cannot download in the Nix sandbox
# itself # and `dataset` is an optional dependency for transformers
# itself, we will just remove the tests files that import
# `dataset`.
rm tests/test_retrieval_rag.py
rm tests/test_trainer.py rm tests/test_trainer.py
''; '';