[free-threaded] [azure parallel] fix for when category is a tuple

lesteve · lesteve · commit 2dcc36725481 · 2025-08-30T09:47:23.000+02:00
diff --git a/sklearn/utils/parallel.py b/sklearn/utils/parallel.py
@@ -173,7 +173,19 @@ def __call__(self, *args, **kwargs):
                     if this_value is not None and not isinstance(this_value, str):
                         this_warning_filter_dict[special_key] = this_value.pattern
 
-                warnings.filterwarnings(**this_warning_filter_dict, append=True)
+                category = this_warning_filter_dict.get("category", None)
+                # For some reason sometimes category is a tuple (maybe from
+                # warnings.simplefilter or pytest.warns) but filterwarnings
+                # does not support it
+                if isinstance(category, tuple):
+                    for this_category in category:
+                        this_warning_filter_dict_copy = this_warning_filter_dict.copy()
+                        this_warning_filter_dict_copy["category"] = this_category
+                        warnings.filterwarnings(
+                            **this_warning_filter_dict_copy, append=True
+                        )
+                else:
+                    warnings.filterwarnings(**this_warning_filter_dict, append=True)
 
             return self.function(*args, **kwargs)
 
