From f0f8ef063755fb0ff5d38cf89e86117e15f883d8 Mon Sep 17 00:00:00 2001 From: Ethan Glaser Date: Mon, 2 Jun 2025 17:26:32 -0700 Subject: [PATCH] Avoid patch warning when spmd estimator is used --- sklbench/benchmarks/sklearn_estimator.py | 1 + 1 file changed, 1 insertion(+) diff --git a/sklbench/benchmarks/sklearn_estimator.py b/sklbench/benchmarks/sklearn_estimator.py index f9c0a75e..e5e6001e 100644 --- a/sklbench/benchmarks/sklearn_estimator.py +++ b/sklbench/benchmarks/sklearn_estimator.py @@ -395,6 +395,7 @@ def measure_sklearn_estimator( estimator_class.__module__.startswith("daal4py") or estimator_class.__module__.startswith("sklearnex") ) + and "spmd" not in estimator_class.__module__ ) sklearnex_logging_stream = get_sklearnex_logging_stream()