From 5476d3d39818b406788baf554ffb66600c12a583 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Thu, 21 May 2026 12:43:06 +0200
Subject: [PATCH 1/3] python3Packages.gpytorch: relax mpmath

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

diff --git a/pkgs/development/python-modules/gpytorch/default.nix b/pkgs/development/python-modules/gpytorch/default.nix
index a6cacfc63e87..a2256c02a1ce 100644
--- a/pkgs/development/python-modules/gpytorch/default.nix
+++ b/pkgs/development/python-modules/gpytorch/default.nix
@@ -31,6 +31,8 @@ buildPythonPackage (finalAttrs: {
     hash = "sha256-1CavS+qrV8YqnsT87GjmJV2LOtvExFYQE5YpYZEw9ts=";
   };
 
+  pythonRelaxDeps = [ "mpmath" ];
+
   build-system = [
     setuptools
     setuptools-scm

From c4f37e2b31e063863d78474396ffc2bcf3da96d2 Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Fri, 22 May 2026 00:57:23 +0200
Subject: [PATCH 2/3] python3Packages.ax-platform: disable failing test

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

diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index 42022b15ebe9..187cb2216951 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -119,6 +119,8 @@ buildPythonPackage (finalAttrs: {
     "test_convert_observations"
     # broken with sqlalchemy 2
     "test_sql_storage"
+    # AssertionError
+    "test_online"
   ]
   ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [
     # flaky

From 951a1ddcf9b6dc13c576d7d0ba2a34ca7810689a Mon Sep 17 00:00:00 2001
From: Fabian Affolter <mail@fabian-affolter.ch>
Date: Fri, 22 May 2026 12:16:49 +0200
Subject: [PATCH 3/3] python3Packages.ax-platform: disable tests with are
 time-sensitive

---
 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 187cb2216951..a82550a8de9c 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -121,6 +121,9 @@ buildPythonPackage (finalAttrs: {
     "test_sql_storage"
     # AssertionError
     "test_online"
+    # Timeout
+    "test_efficient_loo_cv_with_fully_bayesian_model"
+    "test_fitting_auxiliary_experiment_dataset"
   ]
   ++ lib.optionals (stdenv.hostPlatform.isDarwin && stdenv.hostPlatform.isx86_64) [
     # flaky