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