pkgs/development/python-modules/mnist/default.nix

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