1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, numpy
5, scipy
6, numba
7, pandas
8, dask
9, distributed
10, coverage
11, flake8
12, black
13, pytest
14, codecov
15}:
16
17buildPythonPackage rec {
18 pname = "stumpy";
19 version = "1.9.2";
20
21 src = fetchFromGitHub {
22 owner = "TDAmeritrade";
23 repo = "stumpy";
24 rev = "v${version}";
25 sha256 = "0x5kac8fqsi3fkfwjdn0d7anslprxaz6cizky9cyj0rpbp0b0yc3";
26 };
27
28 propagatedBuildInputs = [
29 numpy
30 scipy
31 numba
32 ];
33
34 checkInputs = [
35 pandas
36 dask
37 distributed
38 coverage
39 flake8
40 black
41 pytest
42 codecov
43 ];
44
45 # ignore changed numpy operations
46 checkPhase = ''
47 pytest -k 'not allc'
48 '';
49
50 meta = with lib; {
51 description = "A powerful and scalable library that can be used for a variety of time series data mining tasks";
52 homepage = "https://github.com/TDAmeritrade/stumpy";
53 license = licenses.bsd3;
54 maintainers = [ maintainers.costrouc ];
55 };
56}