tjh.dev
nixpkgs mirror (for testing)
github.com/NixOS/nixpkgs
nix
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, numpy
5, scipy
6, numba
7, pandas
8, dask
9, distributed
10, pytestCheckHook
11}:
12
13buildPythonPackage rec {
14 pname = "stumpy";
15 version = "1.10.2";
16
17 src = fetchFromGitHub {
18 owner = "TDAmeritrade";
19 repo = "stumpy";
20 rev = "v${version}";
21 sha256 = "1v17lxqgvkd3n33c2y1j1zy74xy92vsx2l89yhan89msnnb7aafr";
22 };
23
24 propagatedBuildInputs = [
25 numpy
26 scipy
27 numba
28 ];
29
30 checkInputs = [
31 pandas
32 dask
33 distributed
34 pytestCheckHook
35 ];
36
37 pytestFlagsArray = [
38 # whole testsuite is very CPU intensive, only run core tests
39 # TODO: move entire test suite to passthru.tests
40 "tests/test_core.py"
41 ];
42
43 meta = with lib; {
44 description = "A powerful and scalable library that can be used for a variety of time series data mining tasks";
45 homepage = "https://github.com/TDAmeritrade/stumpy";
46 license = licenses.bsd3;
47 maintainers = [ maintainers.costrouc ];
48 };
49}