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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5   setuptools,
   6   setuptools-scm,
   7   numpy,
   8   scipy,
   9   numba,
  10   pandas,
  11   dask,
  12   distributed,
  13   pytestCheckHook,
  14   pythonOlder,
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "stumpy";
  19   version = "1.13.0";
  20   pyproject = true;
  21
  22   disabled = pythonOlder "3.7";
  23
  24   src = fetchFromGitHub {
  25     owner = "TDAmeritrade";
  26     repo = "stumpy";
  27     tag = "v${version}";
  28     hash = "sha256-S+Rb6pHphXfbqz4VMnN1p7ZrlWB/g7XCdy/T5/Q8VD8=";
  29   };
  30
  31   build-system = [
  32     setuptools
  33     setuptools-scm
  34   ];
  35
  36   dependencies = [
  37     numpy
  38     scipy
  39     numba
  40   ];
  41
  42   nativeCheckInputs = [
  43     pandas
  44     dask
  45     distributed
  46     pytestCheckHook
  47   ];
  48
  49   pythonImportsCheck = [ "stumpy" ];
  50
  51   pytestFlagsArray = [
  52     # whole testsuite is very CPU intensive, only run core tests
  53     # TODO: move entire test suite to passthru.tests
  54     "tests/test_core.py"
  55   ];
  56
  57   meta = {
  58     description = "Library that can be used for a variety of time series data mining tasks";
  59     changelog = "https://github.com/TDAmeritrade/stumpy/blob/${src.rev}/CHANGELOG.md";
  60     homepage = "https://github.com/TDAmeritrade/stumpy";
  61     license = lib.licenses.bsd3;
  62     maintainers = [ ];
  63   };
  64 }