pkgs/development/python-modules/tbats/default.nix

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , setuptools
   5 , numpy
   6 , pmdarima
   7 , scikit-learn
   8 , scipy
   9 , pytestCheckHook
  10 }:
  11
  12 buildPythonPackage 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 }