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

   1 {
   2   buildPythonPackage,
   3   fetchFromGitHub,
   4   isPy27,
   5   lib,
   6   mock,
   7   numpy,
   8   pytestCheckHook,
   9 }:
  10
  11 buildPythonPackage 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 }