1{
2 lib,
3 stdenv,
4 buildPythonPackage,
5 fetchFromGitHub,
6
7 # build-system
8 poetry-core,
9
10 # dependencies
11 scikit-learn,
12 numpy,
13 scipy,
14 colorama,
15
16 # tests
17 jupyter,
18 matplotlib,
19 nbconvert,
20 nbformat,
21 pytestCheckHook,
22}:
23
24buildPythonPackage rec {
25 pname = "bayesian-optimization";
26 version = "2.0.3";
27 pyproject = true;
28
29 src = fetchFromGitHub {
30 owner = "bayesian-optimization";
31 repo = "BayesianOptimization";
32 tag = "v${version}";
33 hash = "sha256-vT8MlfAdzIKj6uyQedYngP6rCkIZwS8EdtKs4+8l9CA=";
34 };
35
36 build-system = [ poetry-core ];
37
38 dependencies = [
39 scikit-learn
40 numpy
41 scipy
42 colorama
43 ];
44
45 nativeCheckInputs = [
46 jupyter
47 matplotlib
48 nbconvert
49 nbformat
50 pytestCheckHook
51 ];
52
53 pythonImportsCheck = [ "bayes_opt" ];
54
55 __darwinAllowLocalNetworking = true;
56
57 meta = {
58 description = "Python implementation of global optimization with gaussian processes";
59 homepage = "https://github.com/bayesian-optimization/BayesianOptimization";
60 changelog = "https://github.com/bayesian-optimization/BayesianOptimization/releases/tag/v${version}";
61 license = lib.licenses.mit;
62 maintainers = [ lib.maintainers.juliendehos ];
63 };
64}