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