From 71b3dbc799da77a6c7124f198852c3d6353c04a2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Dani=C3=ABl=20de=20Kok?= <me@danieldk.eu>
Date: Wed, 30 Sep 2020 10:35:48 +0200
Subject: [PATCH] python3Packages.transformers: 3.2.0 -> 3.3.1

Changes:

https://github.com/huggingface/transformers/releases/tag/v3.3.0
https://github.com/huggingface/transformers/releases/tag/v3.3.1
---
 .../python-modules/transformers/default.nix         | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/transformers/default.nix b/pkgs/development/python-modules/transformers/default.nix
index 328cfb0ba8b..fdb807bd0a1 100644
--- a/pkgs/development/python-modules/transformers/default.nix
+++ b/pkgs/development/python-modules/transformers/default.nix
@@ -17,13 +17,13 @@
 
 buildPythonPackage rec {
   pname = "transformers";
-  version = "3.2.0";
+  version = "3.3.1";
 
   src = fetchFromGitHub {
     owner = "huggingface";
     repo = pname;
     rev = "v${version}";
-    sha256 = "0jj94153kgdyklra30xcszxv11hwzfigzy82fgvgzvbwlxv3a1j5";
+    sha256 = "1j9nzhl0zw5z9rnvzfih7v6bax353rxp05b3f0cvkii3b5dbkc2j";
   };
 
   propagatedBuildInputs = [
@@ -52,9 +52,12 @@ buildPythonPackage rec {
   preCheck = ''
     export HOME="$TMPDIR"
 
-    # This test requires the nlp module, which we haven't
-    # packaged yet. However, nlp is optional for transformers
-    # itself
+    # This test requires the `datasets` module to download test
+    # data. However, since we cannot download in the Nix sandbox
+    # 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
   '';