python3Packages.transformers: 3.1.0 -> 3.2.0

, regex

, requests

, numpy

, parameterized

, sacremoses

, sentencepiece

, timeout-decorator

buildPythonPackage rec {

  pname = "transformers";

  version = "3.1.0";

  version = "3.2.0";

  src = fetchFromGitHub {

    owner = "huggingface";

    repo = pname;

    rev = "v${version}";

    sha256 = "0wg36qrcljmpsyhjaxpqw3s1r6276yg8cq0bjrf52l4zlc5k4xzk";

    sha256 = "0jj94153kgdyklra30xcszxv11hwzfigzy82fgvgzvbwlxv3a1j5";

  };

  propagatedBuildInputs = [

  ];

  checkInputs = [

    parameterized

    pytestCheckHook

    timeout-decorator

  ];

  preCheck = ''

    export HOME="$TMPDIR"

    cd tests

    # This test requires the nlp module, which we haven't

    # packaged yet. However, nlp is optional for transformers

    # itself

    rm test_trainer.py

    rm tests/test_trainer.py

  '';

  # We have to run from the main directory for the tests. However,

  # letting pytest discover tests leads to errors.

  pytestFlagsArray = [ "tests" ];

  # Disable tests that require network access.

  disabledTests = [

    "PegasusTokenizationTest"

    "test_tokenizer_from_model_type"

    "test_tokenizer_from_model_type"

    "test_tokenizer_from_pretrained"

    "test_tokenizer_from_tokenizer_class"

    "test_tokenizer_identifier_with_correct_config"

  ];