From 4ca551bbe89c6e06430f8ad49eb7dfe6c802d087 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Fri, 3 Oct 2025 16:40:15 +0000
Subject: [PATCH 1/2] python3Packages.botorch: add optional-dependencies.pymoo

---
 .../python-modules/botorch/default.nix        | 21 ++++++++++++++++---
 1 file changed, 18 insertions(+), 3 deletions(-)

diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix
index 4253117e7563..f2b60e3b4a26 100644
--- a/pkgs/development/python-modules/botorch/default.nix
+++ b/pkgs/development/python-modules/botorch/default.nix
@@ -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
   ];

From 73e045c99a45d5f53bb1cfbce372299cab03c1da Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Thu, 2 Oct 2025 18:09:59 +0000
Subject: [PATCH 2/2] python3Packages.ax-platform: 1.0.0 -> 1.1.2

Diff: https://github.com/facebook/ax/compare/1.0.0...1.1.2

Changelog: https://github.com/facebook/Ax/releases/tag/1.1.2
---
 .../development/python-modules/ax-platform/default.nix | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index 771c02cb87d8..ca6562db40ca 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -29,14 +29,14 @@
 
 buildPythonPackage rec {
   pname = "ax-platform";
-  version = "1.0.0";
+  version = "1.1.2";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "facebook";
     repo = "ax";
     tag = version;
-    hash = "sha256-DFsV1w6J7bTZNUq9OYExDvfc7IfTcthGKAnRMNujRKI=";
+    hash = "sha256-DHB/BcD913gano5N7dlAbB+Tfg1dMTRP3AR1HwJjkLg=";
   };
 
   env.ALLOW_BOTORCH_LATEST = "1";
@@ -57,11 +57,11 @@ buildPythonPackage rec {
     scikit-learn
     scipy
     sympy
-  ];
+  ]
+  ++ botorch.optional-dependencies.pymoo;
 
   nativeCheckInputs = [
     pyfakefs
-    # pytest-xdist
     pytestCheckHook
     sqlalchemy
     tabulate
@@ -74,7 +74,7 @@ buildPythonPackage rec {
     # broken with sqlalchemy 2
     "ax/core/tests/test_experiment.py"
     "ax/service/tests/test_ax_client.py"
-    "ax/service/tests/test_scheduler.py"
+    "ax/service/tests/test_orchestrator.py"
     "ax/service/tests/test_with_db_settings_base.py"
 
     # Hangs forever