mirror of
https://github.com/NixOS/nixpkgs.git
synced 2026-06-09 23:03:47 +00:00
40 lines
887 B
Nix
40 lines
887 B
Nix
{
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
lib,
|
|
numpy,
|
|
pytestCheckHook,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "mnist";
|
|
version = "0.2.2";
|
|
format = "setuptools";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "datapythonista";
|
|
repo = "mnist";
|
|
rev = "${pname}-${version}";
|
|
sha256 = "17r37pbxiv5dw857bmg990x836gq6sgww069w3q5jjg9m3xdm7dh";
|
|
};
|
|
|
|
propagatedBuildInputs = [ numpy ];
|
|
|
|
nativeCheckInputs = [ pytestCheckHook ];
|
|
|
|
# disable tests which fail due to socket related errors
|
|
disabledTests = [
|
|
"test_test_images_has_right_size"
|
|
"test_test_labels_has_right_size"
|
|
"test_train_images_has_right_size"
|
|
"test_train_labels_has_right_size"
|
|
];
|
|
|
|
meta = {
|
|
description = "Python utilities to download and parse the MNIST dataset";
|
|
homepage = "https://github.com/datapythonista/mnist";
|
|
license = lib.licenses.bsd3;
|
|
maintainers = [ ];
|
|
};
|
|
}
|