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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , numpy
   5 , scipy
   6 , numba
   7 , pandas
   8 , dask
   9 , distributed
  10 , pytestCheckHook
  11 , pythonOlder
  12 }:
  13
  14 buildPythonPackage rec {
  15   pname = "stumpy";
  16   version = "1.11.1";
  17   format = "setuptools";
  18
  19   disabled = pythonOlder "3.7";
  20
  21   src = fetchFromGitHub {
  22     owner = "TDAmeritrade";
  23     repo = "stumpy";
  24     rev = "refs/tags/v${version}";
  25     hash = "sha256-ARpXqZpWkbvIEDVkxA1SwlWoxq+3WO6tvv/e7WZ/25c=";
  26   };
  27
  28   propagatedBuildInputs = [
  29     numpy
  30     scipy
  31     numba
  32   ];
  33
  34   checkInputs = [
  35     pandas
  36     dask
  37     distributed
  38     pytestCheckHook
  39   ];
  40
  41   pythonImportsCheck = [
  42     "stumpy"
  43   ];
  44
  45   pytestFlagsArray = [
  46     # whole testsuite is very CPU intensive, only run core tests
  47     # TODO: move entire test suite to passthru.tests
  48     "tests/test_core.py"
  49   ];
  50
  51   meta = with lib; {
  52     description = "Library that can be used for a variety of time series data mining tasks";
  53     homepage = "https://github.com/TDAmeritrade/stumpy";
  54     license = licenses.bsd3;
  55     maintainers = with maintainers; [ costrouc ];
  56   };
  57 }