You’re absolutely right. Thank you. The solution given by @sgugger post
get_preds(ds_type= DatasetType.Fix )
get_preds(ds_type=
)