Skip to content

Commit beb6eef

Browse files
effigiesjhlegarreta
authored andcommitted
type: Ignore warning related to bad upstream stubs
1 parent cf64333 commit beb6eef

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

scripts/dwi_gp_estimation_error_analysis.py

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,14 @@ def cross_validate(
7474
"""
7575

7676
rkf = RepeatedKFold(n_splits=cv, n_repeats=n_repeats)
77-
scores = cross_val_score(gpr, X, y, scoring="neg_root_mean_squared_error", cv=rkf)
77+
# scikit-learn stubs do not recognize rkf as a BaseCrossValidator
78+
scores = cross_val_score(
79+
gpr,
80+
X,
81+
y,
82+
scoring="neg_root_mean_squared_error",
83+
cv=rkf, # type: ignore[arg-type]
84+
)
7885
return scores
7986

8087

0 commit comments

Comments
 (0)