depot/third_party/nixpkgs/pkgs/development/python-modules/pymc3/default.nix
Default email 81047829ea Project import generated by Copybara.
GitOrigin-RevId: 48d63e924a2666baf37f4f14a18f19347fbd54a2
2022-02-10 15:34:41 -05:00

75 lines
1.6 KiB
Nix

{ lib
, fetchPypi
, buildPythonPackage
, pythonOlder
, Theano
, pandas
, patsy
, joblib
, tqdm
, six
, h5py
, arviz
, packaging
, pytest
, nose
, parameterized
, fastprogress
, typing-extensions
}:
buildPythonPackage rec {
pname = "pymc3";
version = "3.11.4";
disabled = pythonOlder "3.5";
src = fetchPypi {
inherit pname version;
sha256 = "3b88d1e6c85f7fb8a9b99d6f136ac860672170370ec4146338fdd160c3b3fd3f";
};
# No need for coverage stats in Nix builds
postPatch = ''
substituteInPlace setup.py --replace ", 'pytest-cov'" ""
'';
propagatedBuildInputs = [
Theano
pandas
patsy
joblib
tqdm
six
h5py
arviz
packaging
fastprogress
typing-extensions
];
checkInputs = [
pytest
nose
parameterized
];
# The test suite is computationally intensive and test failures are not
# indicative for package usability hence tests are disabled by default.
doCheck = false;
pythonImportsCheck = [ "pymc3" ];
# For some reason tests are run as a part of the *install* phase if enabled.
# Theano writes compiled code to ~/.theano hence we set $HOME.
preInstall = "export HOME=$(mktemp -d)";
postInstall = "rm -rf $HOME";
meta = {
description = "Bayesian estimation, particularly using Markov chain Monte Carlo (MCMC)";
homepage = "https://github.com/pymc-devs/pymc3";
license = lib.licenses.asl20;
maintainers = with lib.maintainers; [ ilya-kolpakov ];
# several dependencies are not declared and in the end it requires theano-pymc3
# instead of Theano. The former is currently not packaged.
broken = true;
};
}