Skip to content

Commit 3d6c08f

Browse files
don't fallback for probability=deprecated (#3038)
1 parent d555b74 commit 3d6c08f

1 file changed

Lines changed: 3 additions & 1 deletion

File tree

sklearnex/svm/_classes.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -138,7 +138,9 @@ def _onedal_gpu_supported(self, method_name, *data):
138138
# support for CalibratedClassifierCV with the arguments used here.
139139
(
140140
not (
141-
self.probability
141+
hasattr(self, "probability")
142+
and self.probability
143+
and self.probability != "deprecated"
142144
and hasattr(X, "__dlpack__")
143145
and not isinstance(X, np.ndarray)
144146
),

0 commit comments

Comments
 (0)