Update progressive_validation.py

online-ml · Oct 30, 2023 · b9f7366 · b9f7366
1 parent c4afffc
commit b9f7366
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/river/evaluate/progressive_validation.py b/river/evaluate/progressive_validation.py
@@ -190,6 +190,8 @@ def iter_progressive_val_score(
 
     The `yield_predictions` parameter can be used to include the predictions in the results:
 
+    >>> import itertools
+
     >>> steps = evaluate.iter_progressive_val_score(
     ...     model=model,
     ...     dataset=datasets.Phishing(),