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 # disable tests which fail due to socket related errors
28 disabledTests = [
29 "test_test_images_has_right_size"
30 "test_test_labels_has_right_size"
31 "test_train_images_has_right_size"
32 "test_train_labels_has_right_size"
33 ];
34
35 meta = with lib; {
36 description = "Python utilities to download and parse the MNIST dataset";
37 homepage = "https://github.com/datapythonista/mnist";
38 license = licenses.bsd3;
39 maintainers = with maintainers; [ ];
40 };
41}