diff options
Diffstat (limited to 'pkgs/development/python-modules/tbats/default.nix')
-rw-r--r-- | pkgs/development/python-modules/tbats/default.nix | 56 |
1 files changed, 56 insertions, 0 deletions
diff --git a/pkgs/development/python-modules/tbats/default.nix b/pkgs/development/python-modules/tbats/default.nix new file mode 100644 index 00000000000..8976fef859d --- /dev/null +++ b/pkgs/development/python-modules/tbats/default.nix @@ -0,0 +1,56 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, setuptools +, numpy +, pmdarima +, scikit-learn +, scipy +, pytestCheckHook +}: + +buildPythonPackage rec { + pname = "tbats"; + version = "1.1.3"; + pyproject = true; + + src = fetchFromGitHub { + owner = "intive-DataScience"; + repo = "tbats"; + rev = version; + hash = "sha256-f6QqDq/ffbnFBZRAT6KQRlqvZZSE+Pff2/o+htVabZI="; + }; + + nativeBuildInputs = [ + setuptools + ]; + + propagatedBuildInputs = [ + numpy + pmdarima + scikit-learn + scipy + ]; + + nativeCheckInputs = [ pytestCheckHook ]; + pytestFlagsArray = [ + # test_R folder is just for comparison of results with R lib + # we need only test folder + "test/" + + # several tests has same name, so we use --deselect instead of disableTests + + # Test execution is too long > 15 min + "--deselect=test/tbats/TBATS_test.py::TestTBATS::test_fit_predict_trigonometric_seasonal" + ]; + + pythonImportsCheck = [ "tbats" ]; + + meta = with lib; { + description = "BATS and TBATS forecasting methods"; + homepage = "https://github.com/intive-DataScience/tbats"; + changelog = "https://github.com/intive-DataScience/tbats/releases/tag/${src.rev}"; + license = licenses.mit; + maintainers = with maintainers; [ mbalatsko ]; + }; +} |