pyrox.dev
1{ lib
2, buildPythonPackage
3, fetchFromGitHub
4, setuptools
5, numpy
6, pmdarima
7, scikit-learn
8, scipy
9, pytestCheckHook
10}:
11
12buildPythonPackage rec {
13 pname = "tbats";
14 version = "1.1.3";
15 pyproject = true;
16
17 src = fetchFromGitHub {
18 owner = "intive-DataScience";
19 repo = "tbats";
20 rev = version;
21 hash = "sha256-f6QqDq/ffbnFBZRAT6KQRlqvZZSE+Pff2/o+htVabZI=";
22 };
23
24 nativeBuildInputs = [
25 setuptools
26 ];
27
28 propagatedBuildInputs = [
29 numpy
30 pmdarima
31 scikit-learn
32 scipy
33 ];
34
35 nativeCheckInputs = [ pytestCheckHook ];
36 pytestFlagsArray = [
37 # test_R folder is just for comparison of results with R lib
38 # we need only test folder
39 "test/"
40
41 # several tests has same name, so we use --deselect instead of disableTests
42
43 # Test execution is too long > 15 min
44 "--deselect=test/tbats/TBATS_test.py::TestTBATS::test_fit_predict_trigonometric_seasonal"
45 ];
46
47 pythonImportsCheck = [ "tbats" ];
48
49 meta = with lib; {
50 description = "BATS and TBATS forecasting methods";
51 homepage = "https://github.com/intive-DataScience/tbats";
52 changelog = "https://github.com/intive-DataScience/tbats/releases/tag/${src.rev}";
53 license = licenses.mit;
54 maintainers = with maintainers; [ mbalatsko ];
55 };
56}