Merge pull request #331857 from mweinelt/botorch-stuck-tests

{

  lib,

  stdenv,

  buildPythonPackage,

  fetchFromGitHub,

  gpytorch,

  nativeCheckInputs = [ pytestCheckHook ];

  disabledTests = [ "test_all_cases_covered" ];

  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";