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 packaging,
16
17 # tests
18 jupyter,
19 matplotlib,
20 nbconvert,
21 nbformat,
22 pytestCheckHook,
23}:
24
25buildPythonPackage rec {
26 pname = "bayesian-optimization";
27 version = "3.0.1";
28 pyproject = true;
29
30 src = fetchFromGitHub {
31 owner = "bayesian-optimization";
32 repo = "BayesianOptimization";
33 tag = "v${version}";
34 hash = "sha256-dq5R0/gqjSzQPAmYvtByJ6gT8pOiXcezfYlKpFLnryk=";
35 };
36
37 build-system = [ poetry-core ];
38
39 dependencies = [
40 scikit-learn
41 numpy
42 scipy
43 colorama
44 packaging
45 ];
46
47 nativeCheckInputs = [
48 jupyter
49 matplotlib
50 nbconvert
51 nbformat
52 pytestCheckHook
53 ];
54
55 pythonImportsCheck = [ "bayes_opt" ];
56
57 __darwinAllowLocalNetworking = true;
58
59 meta = {
60 description = "Python implementation of global optimization with gaussian processes";
61 homepage = "https://github.com/bayesian-optimization/BayesianOptimization";
62 changelog = "https://github.com/bayesian-optimization/BayesianOptimization/releases/tag/${src.tag}";
63 license = lib.licenses.mit;
64 maintainers = [ lib.maintainers.juliendehos ];
65 };
66}