1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, numpy
5, scipy
6, numba
7, pandas
8, dask
9, distributed
10, pytestCheckHook
11, pythonOlder
12}:
13
14buildPythonPackage rec {
15 pname = "stumpy";
16 version = "1.12.0";
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-rVl3tIx8iWx2mnaix3V5YnfWWdPBTP8+K2JJKTfctDA=";
26 };
27
28 propagatedBuildInputs = [
29 numpy
30 scipy
31 numba
32 ];
33
34 nativeCheckInputs = [
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; [ ];
56 };
57}