diff --git a/pkgs/development/python-modules/pmdarima/default.nix b/pkgs/development/python-modules/pmdarima/default.nix new file mode 100644 index 000000000000..17bde3ddf3c7 --- /dev/null +++ b/pkgs/development/python-modules/pmdarima/default.nix @@ -0,0 +1,64 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, cython +, joblib +, numpy +, pandas +, scikit-learn +, scipy +, statsmodels +, urllib3 +, pythonOlder +, python +, pytestCheckHook +}: + +buildPythonPackage rec { + pname = "pmdarima"; + version = "2.0.3"; + format = "setuptools"; + + disable = pythonOlder "3.7"; + + src = fetchFromGitHub { + owner = "alkaline-ml"; + repo = "pmdarima"; + rev = "v${version}"; + hash = "sha256-uX4iZZ2deYqVWnqVZT6J0Djf2pXo7ug4MsOsPkKjvSU="; + }; + + nativeBuildInputs = [ cython ]; + + propagatedBuildInputs = [ + joblib + numpy + pandas + scikit-learn + scipy + statsmodels + urllib3 + ]; + + # Make sure we're running the tests for the actually installed + # package, so that cython's compiled files are available. + preCheck = '' + cd $out/lib/${python.libPrefix}/site-packages + ''; + + nativeCheckInputs = [ pytestCheckHook ]; + disabledTests= [ + # touches internet + "test_load_from_web" + ]; + + pythonImportsCheck = [ "pmdarima" ]; + + meta = with lib; { + 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"; + homepage = "https://github.com/alkaline-ml/pmdarima"; + changelog = "https://github.com/alkaline-ml/pmdarima/releases/tag/v${version}"; + license = licenses.mit; + maintainers = with maintainers; [ mbalatsko ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 3f187bb87986..9bf338fb5f9c 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -8500,6 +8500,8 @@ self: super: with self; { pmw = callPackage ../development/python-modules/pmw { }; + pmdarima = callPackage ../development/python-modules/pmdarima { }; + pmdsky-debug-py = callPackage ../development/python-modules/pmdsky-debug-py { }; pnglatex = callPackage ../development/python-modules/pnglatex { };