Martin Weinelt
62a8e3308e
The test phase for this package is very CPU intensive and prone to getting stuck when it has to compete for resources.
76 lines
1.5 KiB
Nix
76 lines
1.5 KiB
Nix
{
|
|
lib,
|
|
stdenv,
|
|
buildPythonPackage,
|
|
fetchFromGitHub,
|
|
gpytorch,
|
|
linear-operator,
|
|
multipledispatch,
|
|
pyro-ppl,
|
|
setuptools,
|
|
setuptools-scm,
|
|
torch,
|
|
scipy,
|
|
pytestCheckHook,
|
|
}:
|
|
|
|
buildPythonPackage rec {
|
|
pname = "botorch";
|
|
version = "0.11.3";
|
|
pyproject = true;
|
|
|
|
src = fetchFromGitHub {
|
|
owner = "pytorch";
|
|
repo = "botorch";
|
|
rev = "refs/tags/v${version}";
|
|
hash = "sha256-AtRU5KC8KlkxMCU0OUAHDFK7BsPO3TbRmmzDGV7+yVk=";
|
|
};
|
|
|
|
build-system = [
|
|
setuptools
|
|
setuptools-scm
|
|
];
|
|
|
|
dependencies = [
|
|
gpytorch
|
|
linear-operator
|
|
multipledispatch
|
|
pyro-ppl
|
|
scipy
|
|
torch
|
|
];
|
|
|
|
pythonRelaxDeps = [
|
|
"gpytorch"
|
|
"linear-operator"
|
|
];
|
|
|
|
nativeCheckInputs = [ pytestCheckHook ];
|
|
|
|
pytestFlagsArray = [
|
|
# tests tend to get stuck on busy hosts, increase verbosity to find out
|
|
# which specific tests get stuck
|
|
"-vvv"
|
|
];
|
|
|
|
disabledTests =
|
|
[ "test_all_cases_covered" ]
|
|
++ lib.optionals (stdenv.buildPlatform.system == "x86_64-linux") [
|
|
# stuck tests on hydra
|
|
"test_moo_predictive_entropy_search"
|
|
];
|
|
|
|
pythonImportsCheck = [ "botorch" ];
|
|
|
|
# needs lots of undisturbed CPU time or prone to getting stuck
|
|
requiredSystemFeatures = [ "big-parallel" ];
|
|
|
|
meta = with lib; {
|
|
changelog = "https://github.com/pytorch/botorch/blob/${src.rev}/CHANGELOG.md";
|
|
description = "Bayesian Optimization in PyTorch";
|
|
homepage = "https://botorch.org";
|
|
license = licenses.mit;
|
|
maintainers = with maintainers; [ veprbl ];
|
|
};
|
|
}
|