From 61d0d004f84343bfffb9a631092b21e41adf6ded Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Sat, 20 Jul 2024 15:51:43 -0400
Subject: [PATCH 1/2] python312Packages.botorch: relax gpytorch version
 requirement

and migrate to PEP517 lingo for dependencies.

Co-Authored-By: Martin Weinelt <hexa@darmstadt.ccc.de>
---
 pkgs/development/python-modules/botorch/default.nix | 12 ++++++++----
 1 file changed, 8 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix
index 129facf57de1..06f0922af4b1 100644
--- a/pkgs/development/python-modules/botorch/default.nix
+++ b/pkgs/development/python-modules/botorch/default.nix
@@ -26,13 +26,13 @@ buildPythonPackage rec {
     hash = "sha256-YX/G46U09y/VZuWZhKY8zU0Y+bf0NKumzSGYUWvrq/0=";
   };
 
-  nativeBuildInputs = [
+  build-system = [
     setuptools
     setuptools-scm
     wheel
   ];
 
-  propagatedBuildInputs = [
+  dependenciess = [
     gpytorch
     linear-operator
     multipledispatch
@@ -41,9 +41,13 @@ buildPythonPackage rec {
     torch
   ];
 
-  pythonRelaxDeps = [ "linear-operator" ];
+  pythonRelaxDeps = [
+    "gpytorch"
+    "linear-operator"
+  ];
+
+  nativeCheckInputs = [ pytestCheckHook ];
 
-  checkInputs = [ pytestCheckHook ];
   pythonImportsCheck = [ "botorch" ];
 
   meta = with lib; {

From b2df3187bcfc42f8ab398b3edd84bd3cd881096a Mon Sep 17 00:00:00 2001
From: Dmitry Kalinkin <dmitry.kalinkin@gmail.com>
Date: Sat, 20 Jul 2024 15:51:54 -0400
Subject: [PATCH 2/2] python312Packages.ax: relax botorch version requirement

---
 pkgs/development/python-modules/ax/default.nix | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/pkgs/development/python-modules/ax/default.nix b/pkgs/development/python-modules/ax/default.nix
index dddcc6e1ebeb..7b3cc8f2b1c2 100644
--- a/pkgs/development/python-modules/ax/default.nix
+++ b/pkgs/development/python-modules/ax/default.nix
@@ -49,6 +49,8 @@ buildPythonPackage rec {
     pyre-extensions
   ];
 
+  env.ALLOW_BOTORCH_LATEST = "1";
+
   checkInputs = [
     hypothesis
     mercurial