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

   1 {
   2   lib,
   3   stdenv,
   4   buildPythonPackage,
   5   fetchFromGitHub,
   6   gpytorch,
   7   linear-operator,
   8   multipledispatch,
   9   pyro-ppl,
  10   setuptools,
  11   setuptools-scm,
  12   torch,
  13   scipy,
  14   pytestCheckHook,
  15 }:
  16
  17 buildPythonPackage rec {
  18   pname = "botorch";
  19   version = "0.12.0";
  20   pyproject = true;
  21
  22   src = fetchFromGitHub {
  23     owner = "pytorch";
  24     repo = "botorch";
  25     tag = "v${version}";
  26     hash = "sha256-CKlerCUadObpPq4jQAiFDBOZMHZ4QccAKQz30OFe64E=";
  27   };
  28
  29   build-system = [
  30     setuptools
  31     setuptools-scm
  32   ];
  33
  34   dependencies = [
  35     gpytorch
  36     linear-operator
  37     multipledispatch
  38     pyro-ppl
  39     scipy
  40     torch
  41   ];
  42
  43   pythonRelaxDeps = [
  44     "gpytorch"
  45     "linear-operator"
  46   ];
  47
  48   nativeCheckInputs = [ pytestCheckHook ];
  49
  50   pytestFlagsArray = [
  51     # tests tend to get stuck on busy hosts, increase verbosity to find out
  52     # which specific tests get stuck
  53     "-vvv"
  54   ];
  55
  56   disabledTests =
  57     [ "test_all_cases_covered" ]
  58     ++ lib.optionals (stdenv.buildPlatform.system == "x86_64-linux") [
  59       # stuck tests on hydra
  60       "test_moo_predictive_entropy_search"
  61     ]
  62     ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isAarch64) [
  63       # Numerical error slightly above threshold
  64       # AssertionError: Tensor-likes are not close!
  65       "test_model_list_gpytorch_model"
  66     ];
  67
  68   pythonImportsCheck = [ "botorch" ];
  69
  70   # needs lots of undisturbed CPU time or prone to getting stuck
  71   requiredSystemFeatures = [ "big-parallel" ];
  72
  73   meta = {
  74     changelog = "https://github.com/pytorch/botorch/blob/${src.rev}/CHANGELOG.md";
  75     description = "Bayesian Optimization in PyTorch";
  76     homepage = "https://botorch.org";
  77     license = lib.licenses.mit;
  78     maintainers = with lib.maintainers; [ veprbl ];
  79   };
  80 }