depot/third_party/nixpkgs/pkgs/development/python-modules/bayesian-optimization/default.nix
Default email 7e47f3658e Project import generated by Copybara.
GitOrigin-RevId: 1925c603f17fc89f4c8f6bf6f631a802ad85d784
2024-09-26 11:04:55 +00:00

62 lines
1.3 KiB
Nix

{
stdenv,
lib,
buildPythonPackage,
fetchFromGitHub,
poetry-core,
scikit-learn,
numpy,
scipy,
colorama,
jupyter,
matplotlib,
nbconvert,
nbformat,
pytestCheckHook,
pythonOlder,
}:
buildPythonPackage rec {
pname = "bayesian-optimization";
version = "1.5.1";
pyproject = true;
disabled = pythonOlder "3.9";
src = fetchFromGitHub {
owner = "bayesian-optimization";
repo = "BayesianOptimization";
rev = "refs/tags/v${version}";
hash = "sha256-pDgvdQhlJ5aMRGdi2qXRXVCdJRvrOP/Nr0SSZyHH1WM=";
};
build-system = [ poetry-core ];
propagatedBuildInputs = [
scikit-learn
numpy
scipy
colorama
];
nativeCheckInputs = [
jupyter
matplotlib
nbconvert
nbformat
pytestCheckHook
];
pythonImportsCheck = [ "bayes_opt" ];
meta = with lib; {
broken = stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64;
description = ''
A Python implementation of global optimization with gaussian processes
'';
homepage = "https://github.com/bayesian-optimization/BayesianOptimization";
changelog = "https://github.com/bayesian-optimization/BayesianOptimization/releases/tag/v${version}";
license = licenses.mit;
maintainers = [ maintainers.juliendehos ];
};
}