76 lines
1.4 KiB
Nix
76 lines
1.4 KiB
Nix
{
|
|
lib,
|
|
buildPythonPackage,
|
|
pythonOlder,
|
|
fetchFromGitHub,
|
|
setuptools,
|
|
|
|
cmdstanpy,
|
|
numpy,
|
|
matplotlib,
|
|
pandas,
|
|
holidays,
|
|
tqdm,
|
|
importlib-resources,
|
|
|
|
dask,
|
|
distributed,
|
|
|
|
pytestCheckHook,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "prophet";
|
|
version = "1.1.7";
|
|
pyproject = true;
|
|
|
|
disabled = pythonOlder "3.7";
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "facebook";
|
|
repo = "prophet";
|
|
tag = "v${version}";
|
|
hash = "sha256-94hxpfpZN3yvDUu+kM7Oc2Yu8+z0Gv6zqYRAwKXgHk4=";
|
|
};
|
|
|
|
sourceRoot = "${src.name}/python";
|
|
|
|
env.PROPHET_REPACKAGE_CMDSTAN = "false";
|
|
|
|
nativeBuildInputs = [ setuptools ];
|
|
|
|
propagatedBuildInputs = [
|
|
cmdstanpy
|
|
numpy
|
|
matplotlib
|
|
pandas
|
|
holidays
|
|
tqdm
|
|
importlib-resources
|
|
];
|
|
|
|
optional-dependencies.parallel = [
|
|
dask
|
|
distributed
|
|
]
|
|
++ dask.optional-dependencies.dataframe;
|
|
|
|
preCheck = ''
|
|
# use the generated files from $out for testing
|
|
mv prophet/tests .
|
|
rm -r prophet
|
|
'';
|
|
|
|
nativeCheckInputs = [ pytestCheckHook ];
|
|
|
|
pythonImportsCheck = [ "prophet" ];
|
|
|
|
meta = {
|
|
changelog = "https://github.com/facebook/prophet/releases/tag/${src.tag}";
|
|
description = "Tool for producing high quality forecasts for time series data that has multiple seasonality with linear or non-linear growth";
|
|
homepage = "https://facebook.github.io/prophet/";
|
|
license = lib.licenses.mit;
|
|
maintainers = with lib.maintainers; [ tomasajt ];
|
|
};
|
|
}
|