pkgs/development/python-modules/bayesian-optimization/default.nix

   1 { stdenv
   2 , lib
   3 , buildPythonPackage
   4 , fetchFromGitHub
   5 , scikit-learn
   6 , scipy
   7 , colorama
   8 , pytestCheckHook
   9 , pythonOlder
  10 }:
  11
  12 buildPythonPackage rec {
  13   pname = "bayesian-optimization";
  14   version = "1.4.3";
  15   format = "setuptools";
  16
  17   disabled = pythonOlder "3.7";
  18
  19   src = fetchFromGitHub {
  20     owner = "bayesian-optimization";
  21     repo = "BayesianOptimization";
  22     rev = "refs/tags/v${version}";
  23     hash = "sha256-Bp/ZhVSW5lTGwnsd/doOXu++Gxw/51owCfMm96Qmgd4=";
  24   };
  25
  26   propagatedBuildInputs = [
  27     scikit-learn
  28     scipy
  29     colorama
  30   ];
  31
  32   nativeCheckInputs = [ pytestCheckHook ];
  33
  34   pythonImportsCheck = [ "bayes_opt" ];
  35
  36   meta = with lib; {
  37     broken = stdenv.isLinux && stdenv.isAarch64;
  38     description = ''
  39       A Python implementation of global optimization with gaussian processes
  40     '';
  41     homepage = "https://github.com/bayesian-optimization/BayesianOptimization";
  42     changelog = "https://github.com/bayesian-optimization/BayesianOptimization/releases/tag/v${version}";
  43     license = licenses.mit;
  44     maintainers = [ maintainers.juliendehos ];
  45   };
  46 }