Skip to content

Commit d068d3c

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

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)