b12f/nixpkgs
1
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

python3Packages.skorch: 0.12.1 -> 0.13.0

Browse source 
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
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
pkgs/development/python-modules/skorch/default.nix
View file

@ -1,4 +1,5 @@
{ 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"
];
# tries to import `transformers` and download HuggingFace data
disabledTestPaths = [ "skorch/tests/test_hf.py" ];
disabledTestPaths = [
# 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" ];