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

   1 { lib
   2 , buildPythonPackage
   3 , fetchFromGitHub
   4 , gpytorch
   5 , linear_operator
   6 , multipledispatch
   7 , pyro-ppl
   8 , setuptools
   9 , setuptools-scm
  10 , wheel
  11 , torch
  12 , scipy
  13 , pytestCheckHook
  14 }:
  15
  16 buildPythonPackage rec {
  17   pname = "botorch";
  18   version = "0.9.3";
  19   format = "pyproject";
  20
  21   src = fetchFromGitHub {
  22     owner = "pytorch";
  23     repo = pname;
  24     rev = "refs/tags/v${version}";
  25     hash = "sha256-d8EMmA499Zxyagkqx0JCKMZPwSH4LvBya+raD3v3iZU=";
  26   };
  27
  28   nativeBuildInputs = [
  29     setuptools
  30     setuptools-scm
  31     wheel
  32   ];
  33
  34   propagatedBuildInputs = [
  35     gpytorch
  36     linear_operator
  37     multipledispatch
  38     pyro-ppl
  39     scipy
  40     torch
  41   ];
  42
  43   SETUPTOOLS_SCM_PRETEND_VERSION = version;
  44
  45   checkInputs = [
  46     pytestCheckHook
  47   ];
  48   pythonImportsCheck = [ "botorch" ];
  49
  50   meta = with lib; {
  51     description = "Bayesian Optimization in PyTorch";
  52     homepage = "https://botorch.org";
  53     license = licenses.mit;
  54     maintainers = with maintainers; [ veprbl ];
  55   };
  56 }