From 731f77026bfbc0eb3f0ae8d70076d7d17a0e27af Mon Sep 17 00:00:00 2001
From: Sarah Clark <seclark@nextquestion.net>
Date: Mon, 23 Feb 2026 14:20:08 -0800
Subject: [PATCH] python3Packages.sagemaker-mlflow: disable test that depends
 on older scikit-learn

---
 pkgs/development/python-modules/sagemaker-mlflow/default.nix | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/pkgs/development/python-modules/sagemaker-mlflow/default.nix b/pkgs/development/python-modules/sagemaker-mlflow/default.nix
index 4cb6d51a5fdf..096108afa892 100644
--- a/pkgs/development/python-modules/sagemaker-mlflow/default.nix
+++ b/pkgs/development/python-modules/sagemaker-mlflow/default.nix
@@ -55,6 +55,10 @@ buildPythonPackage rec {
     "test_log_artifact"
     "test_presigned_url"
     "test_presigned_url_with_fields"
+
+    # https://github.com/aws/sagemaker-mlflow/issues/16
+    # TypeError: LogisticRegression.__init__() got an unexpected keyword argument 'multi_class'
+    "test_model_registry"
   ];
 
   meta = {