python3Packages.skorch: 0.12.1 -> 0.13.0

Changelog: https://github.com/skorch-dev/skorch/blob/master/CHANGES.md
This commit is contained in:
natsukium 2023-05-22 21:37:39 +09:00
parent 8c8b03ba9e
commit 7fc30298e4
No known key found for this signature in database
GPG key ID: 9EA45A31DB994C53

View file

@ -1,4 +1,5 @@
{ lib { lib
, stdenv
, buildPythonPackage , buildPythonPackage
, fetchPypi , fetchPypi
, pytestCheckHook , pytestCheckHook
@ -14,11 +15,11 @@
buildPythonPackage rec { buildPythonPackage rec {
pname = "skorch"; pname = "skorch";
version = "0.12.1"; version = "0.13.0";
src = fetchPypi { src = fetchPypi {
inherit pname version; inherit pname version;
hash = "sha256-fjNbNY/Dr7lgVGPrHJTvPGuhyPR6IVS7ohBQMI+J1+k="; hash = "sha256-k9Zs4uqskHLqVHOKK7dIOmBSUmbDpOMuPS9eSdxNjO0=";
}; };
propagatedBuildInputs = [ numpy torch scikit-learn scipy tabulate tqdm ]; propagatedBuildInputs = [ numpy torch scikit-learn scipy tabulate tqdm ];
@ -37,10 +38,19 @@ buildPythonPackage rec {
"test_load_cuda_params_to_cpu" "test_load_cuda_params_to_cpu"
# failing tests # failing tests
"test_pickle_load" "test_pickle_load"
] ++ lib.optionals stdenv.isDarwin [
# there is a problem with the compiler selection
"test_fit_and_predict_with_compile"
]; ];
# tries to import `transformers` and download HuggingFace data disabledTestPaths = [
disabledTestPaths = [ "skorch/tests/test_hf.py" ]; # tries to import `transformers` and download HuggingFace data
"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" ]; pythonImportsCheck = [ "skorch" ];