Unverified Commit 8f13c7d6 authored by Weijia Wang's avatar Weijia Wang Committed by GitHub
Browse files

Merge pull request #233405 from natsukium/python3Packages.skorch/update 

python3Packages.skorch: 0.12.1 -> 0.13.0
parents 40a17eaf 7fc30298

pkgs/development/python-modules/skorch/default.nix

+14 −4
Original line number Diff line number Diff line 
{ lib

, stdenv

, buildPythonPackage

, fetchPypi

, pytestCheckHook
@@ -14,11 +15,11 @@ 


buildPythonPackage rec {

  pname = "skorch";

  version = "0.12.1";

  version = "0.13.0";



  src = fetchPypi {

    inherit pname version;

    hash = "sha256-fjNbNY/Dr7lgVGPrHJTvPGuhyPR6IVS7ohBQMI+J1+k=";

    hash = "sha256-k9Zs4uqskHLqVHOKK7dIOmBSUmbDpOMuPS9eSdxNjO0=";

  };



  propagatedBuildInputs = [ numpy torch scikit-learn scipy tabulate tqdm ];
@@ -37,10 +38,19 @@ buildPythonPackage rec { 
    "test_load_cuda_params_to_cpu"

    # failing tests

    "test_pickle_load"

  ] ++ lib.optionals stdenv.isDarwin [

    # there is a problem with the compiler selection

    "test_fit_and_predict_with_compile"

  ];



  disabledTestPaths = [

    # tries to import `transformers` and download HuggingFace data

  disabledTestPaths = [ "skorch/tests/test_hf.py" ];

    "skorch/tests/test_hf.py"

  ] ++ lib.optionals (stdenv.hostPlatform.system != "x86_64-linux") [

    # torch.distributed is disabled by default in darwin

    # aarch64-linux also failed these tests

    "skorch/tests/test_history.py"

  ];



  pythonImportsCheck = [ "skorch" ];