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

   1 {
   2   lib,
   3   buildPythonPackage,
   4   fetchFromGitHub,
   5
   6   # build-system
   7   setuptools,
   8   versioneer,
   9
  10   # dependencies
  11   arviz,
  12   cachetools,
  13   cloudpickle,
  14   numpy,
  15   pandas,
  16   pytensor,
  17   rich,
  18   scipy,
  19   threadpoolctl,
  20   typing-extensions,
  21 }:
  22
  23 buildPythonPackage rec {
  24   pname = "pymc";
  25   version = "5.17.0";
  26   pyproject = true;
  27
  28   src = fetchFromGitHub {
  29     owner = "pymc-devs";
  30     repo = "pymc";
  31     rev = "refs/tags/v${version}";
  32     hash = "sha256-vElS6f46xVvK+p5/IvjgCI4wMZlBe3Q5ZaQUie1yLJw=";
  33   };
  34
  35   postPatch = ''
  36     substituteInPlace setup.py \
  37       --replace-fail ', "pytest-cov"' ""
  38   '';
  39
  40   build-system = [
  41     setuptools
  42     versioneer
  43   ];
  44
  45   dependencies = [
  46     arviz
  47     cachetools
  48     cloudpickle
  49     numpy
  50     pandas
  51     pytensor
  52     rich
  53     scipy
  54     threadpoolctl
  55     typing-extensions
  56   ];
  57
  58   # The test suite is computationally intensive and test failures are not
  59   # indicative for package usability hence tests are disabled by default.
  60   doCheck = false;
  61
  62   pythonImportsCheck = [ "pymc" ];
  63
  64   meta = {
  65     description = "Bayesian estimation, particularly using Markov chain Monte Carlo (MCMC)";
  66     homepage = "https://github.com/pymc-devs/pymc";
  67     changelog = "https://github.com/pymc-devs/pymc/releases/tag/v${version}";
  68     license = lib.licenses.asl20;
  69     maintainers = with lib.maintainers; [
  70       nidabdella
  71       ferrine
  72     ];
  73   };
  74 }