Commit 3c810023 authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files
parent c4082242
Loading
Loading
Loading
Loading
+12 −3
Original line number Diff line number Diff line
@@ -29,14 +29,14 @@

buildPythonPackage (finalAttrs: {
  pname = "botorch";
  version = "0.17.0";
  version = "0.17.2";
  pyproject = true;

  src = fetchFromGitHub {
    owner = "meta-pytorch";
    repo = "botorch";
    tag = "v${finalAttrs.version}";
    hash = "sha256-NDdXsmVdrTEXMXXVf89EkGXVOYnEcXwtzarB5niTNaw=";
    hash = "sha256-KWkRdgOAhWoVRoBn0XvKhTQX3KPHJrzqWoiaKrYzi7o=";
  };

  build-system = [
@@ -82,10 +82,19 @@ buildPythonPackage (finalAttrs: {
    # RuntimeError: Boolean value of Tensor with more than one value is ambiguous
    "test_optimize_acqf_mixed_binary_only"
  ]
  ++ lib.optionals (stdenv.buildPlatform.system == "x86_64-linux") [
  ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isx86_64) [
    # stuck tests on hydra
    "test_moo_predictive_entropy_search"
  ]
  ++ lib.optionals (stdenv.hostPlatform.isLinux && stdenv.hostPlatform.isAarch64) [
    # RuntimeError: Failed to initialize cpuinfo!
    "test_append_features"
    "test_delattr_ctx"
    "test_input_perturbation"
    "test_module_rollback_ctx"
    "test_parameter_rollback_ctx"
    "test_zero_grad_ctx"
  ]
  ++ lib.optionals (stdenv.hostPlatform.isDarwin) [
    # RuntimeError: required keyword attribute 'value' has the wrong type
    "test_posterior_in_trace_mode"