diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix index c62fef2478ae..d4503c85c988 100644 --- a/pkgs/development/python-modules/botorch/default.nix +++ b/pkgs/development/python-modules/botorch/default.nix @@ -11,6 +11,7 @@ , torch , scipy , pytestCheckHook +, pythonRelaxDepsHook }: buildPythonPackage rec { @@ -26,6 +27,7 @@ buildPythonPackage rec { }; nativeBuildInputs = [ + pythonRelaxDepsHook setuptools setuptools-scm wheel @@ -40,6 +42,10 @@ buildPythonPackage rec { torch ]; + pythonRelaxDeps = [ + "linear-operator" + ]; + checkInputs = [ pytestCheckHook ]; diff --git a/pkgs/development/python-modules/linear-operator/default.nix b/pkgs/development/python-modules/linear-operator/default.nix index 907cb9c44d73..8ebe174d199b 100644 --- a/pkgs/development/python-modules/linear-operator/default.nix +++ b/pkgs/development/python-modules/linear-operator/default.nix @@ -3,10 +3,12 @@ , fetchFromGitHub , jaxtyping , pytestCheckHook +, pythonRelaxDepsHook , scipy , setuptools , setuptools-scm , torch +, typeguard , wheel }: @@ -23,6 +25,7 @@ buildPythonPackage rec { }; nativeBuildInputs = [ + pythonRelaxDepsHook setuptools setuptools-scm wheel @@ -32,6 +35,11 @@ buildPythonPackage rec { jaxtyping scipy torch + typeguard + ]; + + pythonRelaxDeps = [ + "typeguard" ]; pythonImportsCheck = [ "linear_operator" ]; @@ -42,6 +50,7 @@ buildPythonPackage rec { disabledTests = [ # flaky numerical tests + "test_matmul_matrix_broadcast" "test_svd" ];