diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix new file mode 100644 index 000000000000..ac31c7fdd9a1 --- /dev/null +++ b/pkgs/development/python-modules/botorch/default.nix @@ -0,0 +1,51 @@ +{ lib +, buildPythonPackage +, fetchFromGitHub +, gpytorch +, linear_operator +, multipledispatch +, pyro-ppl +, setuptools-scm +, torch +, scipy +, pytestCheckHook +}: + +buildPythonPackage rec { + pname = "botorch"; + version = "0.8.5"; + format = "pyproject"; + + src = fetchFromGitHub { + owner = "pytorch"; + repo = pname; + rev = "v${version}"; + hash = "sha256-VcNHgfk8OfLJseQxHksycWuCPCudCtOdcRV0XnxHSfU="; + }; + + buildInputs = [ + setuptools-scm + ]; + propagatedBuildInputs = [ + gpytorch + linear_operator + multipledispatch + pyro-ppl + scipy + torch + ]; + + SETUPTOOLS_SCM_PRETEND_VERSION = version; + + checkInputs = [ + pytestCheckHook + ]; + pythonImportsCheck = [ "botorch" ]; + + meta = with lib; { + description = "Bayesian Optimization in PyTorch"; + homepage = "https://botorch.org"; + license = licenses.mit; + maintainers = with maintainers; [ veprbl ]; + }; +} diff --git a/pkgs/top-level/python-packages.nix b/pkgs/top-level/python-packages.nix index 0ce85ba52563..2a26bafde61b 100644 --- a/pkgs/top-level/python-packages.nix +++ b/pkgs/top-level/python-packages.nix @@ -1435,6 +1435,8 @@ self: super: with self; { botocore = callPackage ../development/python-modules/botocore { }; + botorch = callPackage ../development/python-modules/botorch { }; + bottle = callPackage ../development/python-modules/bottle { }; bottleneck = callPackage ../development/python-modules/bottleneck { };