Skip to content

Commit 236d53e

Browse files
fix
1 parent deecbe7 commit 236d53e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

sklearnex/neighbors/common.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -259,7 +259,8 @@ def _onedal_supported(self, device, method_name, *data):
259259
return patching_status
260260

261261
try:
262-
assert_all_finite(data[0], allow_nan=False)
262+
if data[0] is not None:
263+
assert_all_finite(data[0], allow_nan=False)
263264
has_nan = False
264265
except ValueError:
265266
has_nan = True

0 commit comments

Comments
 (0)