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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , numpy
   5 , pytestCheckHook
   6 , pythonOlder
   7 , scipy
   8 , sparse
   9 }:
  10
  11 buildPythonPackage rec {
  12   pname = "tensorly";
  13   version = "0.7.0";
  14   format = "setuptools";
  15
  16   disabled = pythonOlder "3.7";
  17
  18   src = fetchFromGitHub {
  19     owner = pname;
  20     repo = pname;
  21     rev = version;
  22     sha256 = "sha256-VcX3pCczZQUYZaD7xrrkOcj0QPJt28cYTwpZm5D/X3c=";
  23   };
  24
  25   propagatedBuildInputs = [
  26     numpy
  27     scipy
  28     sparse
  29   ];
  30
  31   checkInputs = [
  32     pytestCheckHook
  33   ];
  34
  35   postPatch = ''
  36     # nose is not actually required for anything
  37     # (including testing with the minimal dependencies)
  38     substituteInPlace setup.py \
  39       --replace ", 'nose'" ""
  40   '';
  41
  42   pythonImportsCheck = [
  43     "tensorly"
  44   ];
  45
  46   pytestFlagsArray = [
  47     "tensorly"
  48   ];
  49
  50   disabledTests = [
  51     # AssertionError: Partial_SVD took too long, maybe full_matrices set wrongly
  52     "test_svd_time"
  53   ];
  54
  55   meta = with lib; {
  56     description = "Tensor learning in Python";
  57     homepage = "https://tensorly.org/";
  58     license = licenses.bsd3;
  59     maintainers = with maintainers; [ bcdarwin ];
  60   };
  61 }