Avoid patch warning when spmd estimator is used

ethanglaser · ethanglaser · commit f0f8ef063755 · 2025-06-02T17:26:32.000-07:00
diff --git 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()
 

Original file line number	Diff line number	Diff line change
`@@ -395,6 +395,7 @@ def measure_sklearn_estimator(`
`395`	`395`	`estimator_class.__module__.startswith("daal4py")`
`396`	`396`	`or estimator_class.__module__.startswith("sklearnex")`
`397`	`397`	`)`
	`398`	`+ and "spmd" not in estimator_class.__module__`
`398`	`399`	`)`
`399`	`400`	`sklearnex_logging_stream = get_sklearnex_logging_stream()`
`400`	`401`