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

   1 { lib
   2 , buildPythonPackage
   3 , chainer
   4 , fetchFromGitHub
   5 , jupyter
   6 , keras
   7   #, mxnet
   8 , nbconvert
   9 , nbformat
  10 , nose
  11 , numpy
  12 , parameterized
  13 , pytestCheckHook
  14 , pythonOlder
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "einops";
  19   version = "0.4.1";
  20   format = "setuptools";
  21
  22   disabled = pythonOlder "3.7";
  23
  24   src = fetchFromGitHub {
  25     owner = "arogozhnikov";
  26     repo = pname;
  27     rev = "v${version}";
  28     hash = "sha256-n4R4lcRimuOncisCTs2zJWPlqZ+W2yPkvkWAnx4R91s=";
  29   };
  30
  31   checkInputs = [
  32     chainer
  33     jupyter
  34     keras
  35     # mxnet (has issues with some CPUs, segfault)
  36     nbconvert
  37     nbformat
  38     nose
  39     numpy
  40     parameterized
  41     pytestCheckHook
  42   ];
  43
  44   # No CUDA in sandbox
  45   EINOPS_SKIP_CUPY = 1;
  46
  47   preCheck = ''
  48     export HOME=$(mktemp -d);
  49   '';
  50
  51   pythonImportsCheck = [
  52     "einops"
  53   ];
  54
  55   disabledTests = [
  56     # Tests are failing as mxnet is not pulled-in
  57     # https://github.com/NixOS/nixpkgs/issues/174872
  58     "test_all_notebooks"
  59     "test_dl_notebook_with_all_backends"
  60     "test_backends_installed"
  61   ];
  62
  63   disabledTestPaths = [
  64     "tests/test_layers.py"
  65   ];
  66
  67   meta = with lib; {
  68     description = "Flexible and powerful tensor operations for readable and reliable code";
  69     homepage = "https://github.com/arogozhnikov/einops";
  70     license = licenses.mit;
  71     maintainers = with maintainers; [ yl3dy ];
  72   };
  73 }
  74