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