tjh.dev
1{
2 lib,
3 buildPythonPackage,
4 fetchFromGitHub,
5 numpy,
6 scipy,
7 numba,
8 pandas,
9 dask,
10 distributed,
11 pytestCheckHook,
12 pythonOlder,
13}:
14
15buildPythonPackage rec {
16 pname = "stumpy";
17 version = "1.12.0";
18 format = "setuptools";
19
20 disabled = pythonOlder "3.7";
21
22 src = fetchFromGitHub {
23 owner = "TDAmeritrade";
24 repo = "stumpy";
25 rev = "refs/tags/v${version}";
26 hash = "sha256-rVl3tIx8iWx2mnaix3V5YnfWWdPBTP8+K2JJKTfctDA=";
27 };
28
29 propagatedBuildInputs = [
30 numpy
31 scipy
32 numba
33 ];
34
35 nativeCheckInputs = [
36 pandas
37 dask
38 distributed
39 pytestCheckHook
40 ];
41
42 pythonImportsCheck = [ "stumpy" ];
43
44 pytestFlagsArray = [
45 # whole testsuite is very CPU intensive, only run core tests
46 # TODO: move entire test suite to passthru.tests
47 "tests/test_core.py"
48 ];
49
50 meta = with lib; {
51 description = "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 = [ ];
55 };
56}