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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , cython
   5 , joblib
   6 , matplotlib
   7 , numpy
   8 , pandas
   9 , scikit-learn
  10 , scipy
  11 , statsmodels
  12 , urllib3
  13 , pythonOlder
  14 , python
  15 , pytest7CheckHook
  16 }:
  17
  18 buildPythonPackage rec {
  19   pname = "pmdarima";
  20   version = "2.0.4";
  21   format = "setuptools";
  22
  23   disabled = pythonOlder "3.7";
  24
  25   src = fetchFromGitHub {
  26     owner = "alkaline-ml";
  27     repo = "pmdarima";
  28     rev = "refs/tags/v${version}";
  29     hash = "sha256-LHwPgQRB/vP3hBM8nqafoCrN3ZSRIMWLzqTqDOETOEc=";
  30   };
  31
  32   nativeBuildInputs = [ cython ];
  33
  34   propagatedBuildInputs = [
  35     joblib
  36     numpy
  37     pandas
  38     scikit-learn
  39     scipy
  40     statsmodels
  41     urllib3
  42   ];
  43
  44   # Make sure we're running the tests for the actually installed
  45   # package, so that cython's compiled files are available.
  46   preCheck = ''
  47     cd $out/${python.sitePackages}
  48   '';
  49
  50   nativeCheckInputs = [
  51     matplotlib
  52     pytest7CheckHook
  53   ];
  54
  55   disabledTests= [
  56     # touches internet
  57     "test_load_from_web"
  58   ];
  59
  60   pythonImportsCheck = [ "pmdarima" ];
  61
  62   meta = with lib; {
  63     description = "A statistical library designed to fill the void in Python's time series analysis capabilities, including the equivalent of R's auto.arima function";
  64     homepage = "https://github.com/alkaline-ml/pmdarima";
  65     changelog = "https://github.com/alkaline-ml/pmdarima/releases/tag/v${version}";
  66     license = licenses.mit;
  67     maintainers = with maintainers; [ mbalatsko ];
  68   };
  69 }