From 5a9ec68e45fca43df34235d5787f0efc7a90beab Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 2 Nov 2025 18:03:51 +0000
Subject: [PATCH 1/3] python3Packages.gpytorch: 1.14 -> 1.14.2

Diff: https://github.com/cornellius-gp/gpytorch/compare/v1.14...v1.14.2
---
 .../python-modules/gpytorch/default.nix       | 24 ++++++++++++-------
 1 file changed, 15 insertions(+), 9 deletions(-)

diff --git a/pkgs/development/python-modules/gpytorch/default.nix b/pkgs/development/python-modules/gpytorch/default.nix
index 08db11be0e4f..e6ac8ecadd65 100644
--- a/pkgs/development/python-modules/gpytorch/default.nix
+++ b/pkgs/development/python-modules/gpytorch/default.nix
@@ -2,27 +2,33 @@
   lib,
   buildPythonPackage,
   fetchFromGitHub,
+
+  # build-system
+  setuptools,
+  setuptools-scm,
+
+  # dependencies
   jaxtyping,
   linear-operator,
   mpmath,
   scikit-learn,
   scipy,
-  setuptools,
-  setuptools-scm,
   torch,
+
+  # tests
   pytestCheckHook,
 }:
 
 buildPythonPackage rec {
   pname = "gpytorch";
-  version = "1.14";
+  version = "1.14.2";
   pyproject = true;
 
   src = fetchFromGitHub {
     owner = "cornellius-gp";
     repo = "gpytorch";
     tag = "v${version}";
-    hash = "sha256-whZjqAs3nyjKMzAGi+OnyBtboq0UuV8m11A4IzkWtec=";
+    hash = "sha256-yDIGiA7q4e6T7SdnO+ALcc3ezmJK964T5Nn48+NGJV8=";
   };
 
   build-system = [
@@ -30,8 +36,6 @@ buildPythonPackage rec {
     setuptools-scm
   ];
 
-  pythonRelaxDeps = [ "jaxtyping" ];
-
   dependencies = [
     jaxtyping
     linear-operator
@@ -56,10 +60,12 @@ buildPythonPackage rec {
     "test_t_matmul_matrix"
   ];
 
-  meta = with lib; {
+  meta = {
     description = "Highly efficient and modular implementation of Gaussian Processes, with GPU acceleration";
     homepage = "https://gpytorch.ai";
-    license = licenses.mit;
-    maintainers = with maintainers; [ veprbl ];
+    downloadPage = "https://github.com/cornellius-gp/gpytorch";
+    changelog = "https://github.com/cornellius-gp/gpytorch/releases/tag/${src.tag}";
+    license = lib.licenses.mit;
+    maintainers = with lib.maintainers; [ veprbl ];
   };
 }

From eb8f297921a826a22e80cb0d0f6c7ce404829118 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 2 Nov 2025 18:00:50 +0000
Subject: [PATCH 2/3] python3Packages.botorch: 0.15.1 -> 0.16.0

Diff: https://github.com/pytorch/botorch/compare/v0.15.1...v0.16.0

Changelog: https://github.com/pytorch/botorch/blob/v0.16.0/CHANGELOG.md
---
 .../python-modules/botorch/default.nix          | 17 +++++++++++++----
 1 file changed, 13 insertions(+), 4 deletions(-)

diff --git a/pkgs/development/python-modules/botorch/default.nix b/pkgs/development/python-modules/botorch/default.nix
index f2b60e3b4a26..5cebb2e1e14e 100644
--- a/pkgs/development/python-modules/botorch/default.nix
+++ b/pkgs/development/python-modules/botorch/default.nix
@@ -15,7 +15,9 @@
   pyre-extensions,
   pyro-ppl,
   scipy,
+  threadpoolctl,
   torch,
+  typing-extensions,
 
   # optional-dependencies
   pymoo,
@@ -27,14 +29,14 @@
 
 buildPythonPackage rec {
   pname = "botorch";
-  version = "0.15.1";
+  version = "0.16.0";
   pyproject = true;
 
   src = fetchFromGitHub {
-    owner = "pytorch";
+    owner = "meta-pytorch";
     repo = "botorch";
     tag = "v${version}";
-    hash = "sha256-6hAsKIlwycZtLZn1vkcu4fR85uACA4FSkT5e/wos17A=";
+    hash = "sha256-XpcmWJcKaIxrM79MgjG7IF/DphTH402iltlh8ISeZ64=";
   };
 
   build-system = [
@@ -49,7 +51,9 @@ buildPythonPackage rec {
     pyre-extensions
     pyro-ppl
     scipy
+    threadpoolctl
     torch
+    typing-extensions
   ];
 
   optional-dependencies = {
@@ -62,6 +66,11 @@ buildPythonPackage rec {
     pytestCheckHook
   ];
 
+  disabledTestPaths = [
+    # Requires unpackaged pfns
+    "test_community/models/test_prior_fitted_network.py"
+  ];
+
   disabledTests = [
     "test_all_cases_covered"
 
@@ -93,7 +102,7 @@ buildPythonPackage rec {
   requiredSystemFeatures = [ "big-parallel" ];
 
   meta = {
-    changelog = "https://github.com/pytorch/botorch/blob/${src.tag}/CHANGELOG.md";
+    changelog = "https://github.com/meta-pytorch/botorch/blob/${src.tag}/CHANGELOG.md";
     description = "Bayesian Optimization in PyTorch";
     homepage = "https://botorch.org";
     license = lib.licenses.mit;

From 367082278c70d2659f672c895ec3a6c5f00904c5 Mon Sep 17 00:00:00 2001
From: Gaetan Lepage <gaetan@glepage.com>
Date: Sun, 2 Nov 2025 19:01:16 +0000
Subject: [PATCH 3/3] python3Packages.ax-platform: skip failing test

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

diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index ca6562db40ca..b853ccf3858e 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -86,6 +86,9 @@ buildPythonPackage rec {
     "SQAStoreUtilsTest"
     "SQAStoreTest"
 
+    # ValueError: Expected dim to be an integer greater than or equal to 2. Found dim=1.
+    "test_get_model"
+
     # ValueError: `db_settings` argument should be of type ax.storage.sqa_store
     "test_get_next_trials_with_db"