Commit 4ca551bb authored by Gaetan Lepage's avatar Gaetan Lepage
Browse files

python3Packages.botorch: add optional-dependencies.pymoo

parent fe2a8ef0

pkgs/development/python-modules/botorch/default.nix

+18 −3
Original line number Diff line number Diff line
@@ -3,15 +3,24 @@ 
  stdenv,

  buildPythonPackage,

  fetchFromGitHub,



  # build-system

  setuptools,

  setuptools-scm,



  # dependencies

  gpytorch,

  linear-operator,

  multipledispatch,

  pyre-extensions,

  pyro-ppl,

  setuptools,

  setuptools-scm,

  torch,

  scipy,

  torch,



  # optional-dependencies

  pymoo,



  # tests

  pytestCheckHook,

  pythonAtLeast,

}:
@@ -43,6 +52,12 @@ buildPythonPackage rec { 
    torch

  ];



  optional-dependencies = {

    pymoo = [

      pymoo

    ];

  };



  nativeCheckInputs = [

    pytestCheckHook

  ];