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()