diff options
Diffstat (limited to 'pkgs/development/python-modules/transformers/default.nix')
-rw-r--r-- | pkgs/development/python-modules/transformers/default.nix | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/pkgs/development/python-modules/transformers/default.nix b/pkgs/development/python-modules/transformers/default.nix index 49fbd953d53..bd9040668c0 100644 --- a/pkgs/development/python-modules/transformers/default.nix +++ b/pkgs/development/python-modules/transformers/default.nix @@ -7,6 +7,7 @@ , regex , requests , numpy +, pandas , parameterized , protobuf , sacremoses @@ -18,14 +19,14 @@ buildPythonPackage rec { pname = "transformers"; - version = "4.0.1"; + version = "4.1.1"; disabled = isPy39; src = fetchFromGitHub { owner = "huggingface"; repo = pname; rev = "v${version}"; - sha256 = "1h53a3n6fdrx3ns1h1ip273hzd9pkm9m1qh41si75csb8mi1dq3d"; + sha256 = "1l1gxdsakjmzsgggypq45pnwm87brhlccjfzafs43460pz0wbd6k"; }; propagatedBuildInputs = [ @@ -41,6 +42,7 @@ buildPythonPackage rec { ]; checkInputs = [ + pandas parameterized pytestCheckHook timeout-decorator |