pyrox.dev
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
16buildPythonPackage 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}