Skip to content

Commit ac0e51c

Browse files
committed
More dask==2025.4.1 compatibility
Followup to rapidsai#6614
1 parent 41e397d commit ac0e51c

1 file changed

Lines changed: 5 additions & 3 deletions

File tree

python/cuml/cuml/dask/neighbors/kneighbors_regressor.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -258,9 +258,11 @@ def score(self, X, y):
258258
-------
259259
score
260260
"""
261-
y_pred = self.predict(X, convert_dtype=True)
262-
if not isinstance(y_pred, da.Array):
263-
y_pred = y_pred.to_dask_array(lengths=True)
261+
y_pred_plain = self.predict(X, convert_dtype=True)
262+
if not isinstance(y_pred_plain, da.Array):
263+
y_pred = y_pred_plain.to_dask_array(lengths=True)
264+
else:
265+
y_pred = y_pred_plain
264266
if not isinstance(y, da.Array):
265267
y = y.to_dask_array(lengths=True)
266268
y_true = y.squeeze()

0 commit comments

Comments
 (0)