pyrox.dev
1{
2 buildPythonPackage,
3 fetchFromGitHub,
4 isPy27,
5 lib,
6 mock,
7 numpy,
8 pytestCheckHook,
9}:
10
11buildPythonPackage rec {
12 pname = "mnist";
13 version = "0.2.2";
14 format = "setuptools";
15
16 src = fetchFromGitHub {
17 owner = "datapythonista";
18 repo = "mnist";
19 rev = "${pname}-${version}";
20 sha256 = "17r37pbxiv5dw857bmg990x836gq6sgww069w3q5jjg9m3xdm7dh";
21 };
22
23 propagatedBuildInputs = [ numpy ] ++ lib.optional isPy27 mock;
24
25 nativeCheckInputs = [ pytestCheckHook ];
26
27 dontUseSetuptoolsCheck = true;
28
29 # disable tests which fail due to socket related errors
30 disabledTests = [
31 "test_test_images_has_right_size"
32 "test_test_labels_has_right_size"
33 "test_train_images_has_right_size"
34 "test_train_labels_has_right_size"
35 ];
36
37 meta = with lib; {
38 description = "Python utilities to download and parse the MNIST dataset";
39 homepage = "https://github.com/datapythonista/mnist";
40 license = licenses.bsd3;
41 maintainers = with maintainers; [ dmrauh ];
42 };
43}