tjh.dev
1{ lib
2, buildPythonPackage
3, fetchPypi
4, pytestCheckHook
5, pytest
6, pytestcov
7, flaky
8, numpy
9, pandas
10, pytorch
11, scikit-learn
12, scipy
13, tabulate
14, tqdm
15}:
16
17buildPythonPackage rec {
18 pname = "skorch";
19 version = "0.10.0";
20
21 src = fetchPypi {
22 inherit pname version;
23 sha256 = "9910f97339e654c8d38e0075d87b735e69e5eb11db59c527fb36705b30c8d0a4";
24 };
25
26 propagatedBuildInputs = [ numpy pytorch scikit-learn scipy tabulate tqdm ];
27 checkInputs = [ pytest pytestcov flaky pandas pytestCheckHook ];
28
29 disabledTests = [
30 # on CPU, these expect artifacts from previous GPU run
31 "test_load_cuda_params_to_cpu"
32 # failing tests
33 "test_pickle_load"
34 "test_grid_search_with_slds_"
35 "test_grid_search_with_dict_works"
36 ];
37
38 meta = with lib; {
39 description = "Scikit-learn compatible neural net library using Pytorch";
40 homepage = "https://skorch.readthedocs.io";
41 changelog = "https://github.com/skorch-dev/skorch/blob/master/CHANGES.md";
42 license = licenses.bsd3;
43 maintainers = with maintainers; [ bcdarwin ];
44 # TypeError: __init__() got an unexpected keyword argument 'iid'
45 broken = true;
46 };
47}