diff --git a/pkgs/development/python-modules/ax-platform/default.nix b/pkgs/development/python-modules/ax-platform/default.nix
index 42022b15ebe9..a82550a8de9c 100644
--- a/pkgs/development/python-modules/ax-platform/default.nix
+++ b/pkgs/development/python-modules/ax-platform/default.nix
@@ -119,6 +119,11 @@ buildPythonPackage (finalAttrs: {
     "test_convert_observations"
     # broken with sqlalchemy 2
     "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
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