code: def yield_batch_dataset():
      """Batch dataset."""
      for x, y in yield_dataset():
          yield pd.Series([x]), pd.Series([y])
  
file_path: river/river/naive_bayes/test_naive_bayes.py
imports: from __future__ import annotations
  import pandas
  import pytest
  from sklearn import naive_bayes
  from river import compose, feature_extraction, naive_bayes
label_desc: Use image node_id 4 for calling a global function with example usage: yield_batch_dataset() without return types