depot/third_party/nixpkgs/pkgs/development/python-modules/pymc/default.nix

74 lines
1.3 KiB
Nix

{
lib,
buildPythonPackage,
fetchFromGitHub,
# build-system
setuptools,
versioneer,
# dependencies
arviz,
cachetools,
cloudpickle,
numpy,
pandas,
pytensor,
rich,
scipy,
threadpoolctl,
typing-extensions,
}:
buildPythonPackage rec {
pname = "pymc";
version = "5.18.0";
pyproject = true;
src = fetchFromGitHub {
owner = "pymc-devs";
repo = "pymc";
rev = "refs/tags/v${version}";
hash = "sha256-rH6sh05rSrhakcfdc+BSR37kQYO/QEPuw8hHMZZGwdE=";
};
postPatch = ''
substituteInPlace setup.py \
--replace-fail ', "pytest-cov"' ""
'';
build-system = [
setuptools
versioneer
];
dependencies = [
arviz
cachetools
cloudpickle
numpy
pandas
pytensor
rich
scipy
threadpoolctl
typing-extensions
];
# The test suite is computationally intensive and test failures are not
# indicative for package usability hence tests are disabled by default.
doCheck = false;
pythonImportsCheck = [ "pymc" ];
meta = {
description = "Bayesian estimation, particularly using Markov chain Monte Carlo (MCMC)";
homepage = "https://github.com/pymc-devs/pymc";
changelog = "https://github.com/pymc-devs/pymc/releases/tag/v${version}";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [
nidabdella
ferrine
];
};
}