python3Packages.skorch: 0.12.1 -> 0.13.0
Changelog: https://github.com/skorch-dev/skorch/blob/master/CHANGES.md
This commit is contained in:
parent
8c8b03ba9e
commit
7fc30298e4
|
@ -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"
|
||||
];
|
||||
|
||||
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" ];
|
||||
|
||||
|
|
Loading…
Reference in a new issue