AttributeError: 'DatasetAutoFolds' object has no attribute 'split' code example

Example: data.split(n_folds=5) DatasetAutoFolds' object has no attribute 'split'

from surprise import SVD,Reader,Dataset
from surprise.model_selection import KFold
ratings =  pd.read_csv("/content/ratings_small.csv")
data = Dataset.load_from_df(ratings[['userId','movieId','rating']],reader)
kf = KFold(n_splits=5)
kf.split(data)